列表上的环绕运行窗口

Question

我希望使用“滑动窗口”将列表拆分成相同大小的块，但不是截断列表的末尾，而是绕回，以便最后的块可以分布在列表的开头和结尾。

例如，给定一个列表：

l = [1, 2, 3, 4, 5, 6]

我希望生成大小为n=3的块，如下所示：

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

或者，包含大小为n=2的块的相同列表应按如下方式拆分：

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

该列表可以不被均匀地划分为n个子列表(例如，如果原始列表具有长度7和n=3 -或者除7或1之外的任何值)。与通常情况下一样，四舍五入的值len(l) / n可用于确定分割大小。

This post是相关的，尽管它不会像我需要的那样进行包装。我已经尝试过了，但没有管理任何有用的东西。任何建议都将非常受欢迎！

Answer 1

您可以在从itertools.cycle生成的环绕迭代器上使用itertools.islice

from itertools import cycle, islice
def rolling_chunks(l, n):
    return ([list(islice(cycle(l), i + j, i + j + n)) for j in range(0, len(l), n)] for i in range(n))

因此list(rolling_chunks(l, 3))返回：

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

并且该list(rolling_chunks(l, 2))返回：

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