如何在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中文網其他相關文章！