python - 将列表拆分为单独但重叠的 block

Question

假设我有一个列表 A

A = [1,2,3,4,5,6,7,8,9,10]

我想按以下顺序使用上面的列表创建一个新列表(比如 B)。

B = [[1,2,3], [3,4,5], [5,6,7], [7,8,9], [9,10,]]

即前 3 个数字为 A[0,1,2]，后 3 个数字为 A[2,3,4] 等等。

我相信 numpy 中有一个函数可以进行这种操作。

Answer 1

只需使用 Python 的内置列表理解和列表切片来执行此操作:

>>> A = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
>>> size = 3
>>> step = 2
>>> A = [A[i : i + size] for i in range(0, len(A), step)]

这给了你想要的东西:

>>> A
[[1, 2, 3], [3, 4, 5], [5, 6, 7], [7, 8, 9], [9, 10]]

但是您必须编写几行代码以确保您的代码不会因前所未有的大小/步长值而中断。