>백엔드 개발 >파이썬 튜토리얼 >Python에서 슬라이딩 윈도우 반복을 효율적으로 구현하려면 어떻게 해야 합니까?

Python에서 슬라이딩 윈도우 반복을 효율적으로 구현하려면 어떻게 해야 합니까?

DDD
DDD원래의
2024-12-29 02:37:09230검색

How Can I Efficiently Implement Sliding Window Iteration in Python?

슬라이딩 윈도우 반복 기법

이터러블에 대한 슬라이딩 윈도우 반복의 경우 기본 구현에서는 목록 슬라이싱 및 반복을 사용할 수 있습니다.

def rolling_window(seq, window_size):
    it = iter(seq)
    win = [it.next() for cnt in range(window_size)]  # First window
    yield win
    for e in it:  # Subsequent windows
        win[:-1] = win[1:]
        win[-1] = e
        yield win

효율적이고 우아함 솔루션

더 큰 효율성과 우아함을 위해 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])

위 내용은 Python에서 슬라이딩 윈도우 반복을 효율적으로 구현하려면 어떻게 해야 합니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.