Python加入more_itertools.windowed结果

Question

我有一个以下问题：我正在尝试创建所谓的“digrams”，如下所示：

如果我有一个单词foobar，我想获得一个列表或生成器，如：["fo", "oo", "ob", "ba", "ar"]。完美的功能是more_itertools.windowed。问题是，它返回元组，如下所示：

In [1]: from more_itertools import windowed

In [2]: for i in windowed("foobar", 2):
   ...:     print(i)
   ...:
('f', 'o')
('o', 'o')
('o', 'b')
('b', 'a')
('a', 'r')

当然我知道我可以.join()他们，所以我会：

In [3]: for i in windowed("foobar", 2):
   ...:     print(''.join(i))
   ...:
   ...:
fo
oo
ob
ba
ar

我只是想知道在itertools或more_itertools的某个地方是否有一个功能，我没有看到它确实如此。或者是否有一种更加“pythonic”的手工方式？

Answer 1

您可以使用切片编写自己的widowed版本。

def str_widowed(s, n):
    for i in range(len(s) - n + 1):
        yield s[i:i+n]

这确保了yielding类型与输入相同，但不再接受非索引的iterables。

Answer 2

more_itertools.windowed()是pythonic。考虑pairwise() itertools recipe也产生元组：

def pairwise(iterable):
    "s -> (s0, s1), (s1, s2), (s2, s3), ..."
    a, b = tee(iterable)
    next(b, None)
    return zip(a, b)

您可以使用windowed()轻松替换pairwise()并获得常见结果 - 一般解决方案。

---

或者，您可以通过模拟压缩重复但偏移字符串的成对原则来切割字符串：

码

s = "foobar"

[a + b for a, b in zip(s, s[1:])]
# ['fo', 'oo', 'ob', 'ba', 'ar']