Цикл массива как четыре элемента, но последовательно - PullRequest
0 голосов
/ 11 января 2019

Я использую следующий код для обхода массива.

arr = [1 ,2 ,3 ,4 ,5 , 6,7]
    for a, b, c in zip(*[iter(arr)]*3):
        print (a, b, c)

Получает выходные данные в двух частях как (1,2,3) и (4,5,6)

Однако я хочу, чтобы выходные данные были последовательными в смысле (1,2,3), (2,3,4), (3,4,5), (4,5,6), (5,6 , 7) но и быстрее. Есть ли другой способ, кроме iter, достичь этого?

Ответы [ 3 ]

0 голосов
/ 11 января 2019

Вы также можете зациклить ломтики размером n

n = 3
for a, b, c in [arr[i:i+n] for i in range(len(arr)-(n-1))]:
    print(a, b, c)
#1 2 3
#2 3 4
#3 4 5
#4 5 6
#5 6 7
0 голосов
/ 11 января 2019
from toolz.itertoolz import sliding_window
arr = [1 ,2 ,3 ,4 ,5 , 6,7]
list(sliding_window(3,arr))

выход

[(1, 2, 3), (2, 3, 4), (3, 4, 5), (4, 5, 6), (5, 6, 7)]
0 голосов
/ 11 января 2019

Просто используйте нарезку:

>>> l = list(range(10))
>>> list(zip(l, l[1:], l[2:]))
[(0, 1, 2), (1, 2, 3), (2, 3, 4), (3, 4, 5), (4, 5, 6), (5, 6, 7), (6, 7, 8), (7, 8, 9)]

Будет еще лучше, если вы используете itertools.islice

>>> from itertools import islice
>>> list(zip(l, islice(l, 1, None), islice(l, 2, None)))
[(0, 1, 2), (1, 2, 3), (2, 3, 4), (3, 4, 5), (4, 5, 6), (5, 6, 7), (6, 7, 8), (7, 8, 9)]
...