Comment créer efficacement un itérateur de fenêtre déroulante en Python ?

Itérateurs de fenêtre glissante : une implémentation Python

Itérer sur les données dans une fenêtre glissante (ou fenêtre glissante) s'avère très utile dans diverses données applications d’analyse et d’apprentissage automatique. Cette question présente une technique permettant de créer un itérateur de fenêtre déroulante pour une séquence ou un itérateur donné en Python.

La principale implémentation Python fournie est la suivante :

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

Cette approche parcourt la séquence , créant la fenêtre initiale, puis mettant à jour la fenêtre dans les itérations suivantes. Cependant, certains passionnés de Python suggèrent des méthodes alternatives pour plus d'élégance et d'efficacité.

Une de ces suggestions implique l'utilisation du module 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

Cette méthode exploite le puissant module itertools pour implémenter la fenêtre déroulante. fonctionnalité succinctement.

Une autre alternative, particulièrement adaptée aux listes ou aux tuples, utilise un simple approche :

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

En fonction des exigences spécifiques et des considérations de performances, ces implémentations alternatives peuvent offrir une flexibilité dans la mise en œuvre d'itérateurs de fenêtres glissantes pour diverses applications.

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!