如何在Python中高效地实现滚动或滑动窗口迭代器？

滚动或滑动窗口迭代器

滚动或滑动窗口迭代器允许在指定大小的重叠窗口中遍历序列。迭代窗口大小为 1 的序列本质上是默认的 Python 迭代。

高效而优雅的实现

Python 文档中的以下实现利用了 itertools 模块为了提高效率：

from itertools import islice

def window(seq, n=2):
    """Returns a sliding window (of width n) over data from the iterable"""
    "s -> (s0,s1,...s[n-1]), (s1,s2,...,sn), ..."
    it = iter(seq)
    result = tuple(islice(it, n))
    if len(result) == n:
        yield result
    for elem in it:
        result = result[1:] + (elem,)
        yield result

实现简单可迭代

对于表示为简单列表或元组的可迭代，一种简单的方法是使用指定的窗口大小迭代可迭代：

seq = [0, 1, 2, 3, 4, 5]
window_size = 3

for i in range(len(seq) - window_size + 1):
    print(seq[i: i + window_size])

输出：

[0, 1, 2]
[1, 2, 3]
[2, 3, 4]
[3, 4, 5]

以上是如何在Python中高效地实现滚动或滑动窗口迭代器？的详细内容。更多信息请关注PHP中文网其他相关文章！