移动窗口列表

Question

假设我有列表[1，2，2，2，3，4，5，6，7]我需要产生输出（移动窗口）：

0  1
8  2
16 3
24 4
32 5

0  2
8  3
16 4
24 5
32 6

0  3
8  4
16 5
24 6
32 7

0  1
8  2
16 3
24 4
32 5

当5圈高窗口到达列表的最后一个元素时（这里7）我们需要开始1，我需要像i * 8一样生成第一列。

我带来了这样的东西（但不是完全）：

def w(l):
    global lines
    for i in range(lines - 1, lines + 4):
        location = (i % 5) * 8
        print(f'{location} {l[i]}')
        if lines >= len(l) - 4:
            lines = 1


if __name__ == '__main__':
    ll = [1, 2, 3, 4, 5, 6, 7]
    for i in range(1, 4):
        lines = i
        print(f'-- {i} --')
        w(ll)

输出是：

-- 1 --
0 1
8 2
16 3
24 4
32 5
-- 2 --
8 2
16 3
24 4
32 5
0 6
-- 3 --
16 3
24 4
32 5
0 6
8 7

也许Deque是答案吗？

Answer 1

要修复位置零件，您只需添加枚举并用8次乘以。

def w(l):
    global lines
    for p,i in enumerate(range(lines - 1, lines + 4)):
        location = p * 8
...

Answer 2

您可以使用numpy：

import numpy as np
sliding_window = np.lib.stride_tricks.sliding_window_view([1, 2, 3, 4, 5, 6, 7], 5)
for i in sliding_window:
    for j, k in enumerate(i):
        print(j*8, k)

输出：

0 1
8 2
16 3
24 4
32 5
0 2
8 3
16 4
24 5
32 6
0 3
8 4
16 5
24 6
32 7