python - 在给定的可迭代对象上滑动宽度为 n 的窗口

Question

我有一个序列、窗口大小和步长:

seq = [0,1,2,3,4]
n=4
step=2

from more_itertools import windowed
list(windowed([0,1,2,3,4], n, fillvalue=0, step=step))

结果:

[(0, 1, 2, 3), (2, 3, 4, 0)]

但我需要:

[(0, 1, 2, 3), (2, 3, 4, 0), (4, 0, 0, 0)]

请帮我想个办法

Answer 1

只需编写您自己的windowed 函数:

def windowed(iterable, size, fillvalue=None, step=1):
    for i in range(0, len(iterable), step):
        window = iterable[i:i+size]
        window += [fillvalue] * (size - len(window))
        yield window

>>> list(windowed([0,1,2,3,4], 4, fillvalue=0, step=2))
[[0, 1, 2, 3], [2, 3, 4, 0], [4, 0, 0, 0]]