Comment puis-je implémenter efficacement un itérateur de fenêtre déroulante ou coulissante en Python ?

Itérateur de fenêtre déroulante ou coulissante

Les itérateurs de fenêtre déroulante ou coulissante permettent le parcours d'une séquence dans des fenêtres superposées d'une taille spécifiée. Itérer sur une séquence avec une taille de fenêtre de 1 est essentiellement l'itération Python par défaut.

Implémentation efficace et élégante

L'implémentation suivante de la documentation Python exploite le module itertools pour une efficacité accrue :

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

Mise en œuvre simple Itérables

Pour les itérables représentés sous forme de simples listes ou tuples, une approche simple consiste à parcourir l'itérable avec la taille de fenêtre spécifiée :

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

Sortie :

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

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!