Python 加入 more_itertools.windowed 个结果

Question

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

如果我有一个词 foobar，我想得到一个列表或生成器，例如：["fo", "oo", "ob", "ba", "ar"]。完美的功能是 more_itertools.windowed。问题是，它 returns 元组，像这样：

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]

这确保产生的类型与输入相同，但不再接受非索引迭代。

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']

Python 加入 more_itertools.windowed 个结果

Python join more_itertools.windowed results

python

itertools

python-3.x

more-itertools