如何在Python中高效創建滾動視窗迭代器？

滾動視窗迭代器：Python 實作

以滾動視窗（或滑動視窗）方式迭代資料在各種資料中證明非常有用分析和機器學習應用。這個問題展示了一種在 Python 中為給定序列或迭代器創建滾動視窗迭代器的技術。

提供的主要 Python 實作如下：

def rolling_window(seq, window_size):
    it = iter(seq)
    win = [it.next() for _ in range(window_size)]
    yield win
    for e in it:
        win = win[1:] + (e,)
        yield win

這種方法迭代序列，建立初始窗口，然後在後續迭代中更新視窗。然而，一些 Python 愛好者建議使用替代方法來提高優雅性和效率。

其中一個建議涉及利用 itertools 模組：

from itertools import islice

def window(seq, n=2):
    it = iter(seq)
    result = tuple(islice(it, n))
    if len(result) == n:
        yield result
    for elem in it:
        result = result[1:] + (elem,)
        yield result

該方法利用強大的 itertools 模組來實現滾動視窗功能簡潔。

另一種選擇，特別適合列表或元組，採用簡單的方法：

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

根據具體要求和性能考慮，這些替代實現可以為各種應用程序實現滾動視窗迭代器提供靈活性。

以上是如何在Python中高效創建滾動視窗迭代器？的詳細內容。更多資訊請關注PHP中文網其他相關文章！