只需将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]]
但是您必须写几行代码,以确保代码不会因空前的大小/步长而中断。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)