使用numpy将1D数组按顺序划分为不同大小的块

Question

我想将1D数组按顺序分为2个和3个大小的块 像这样：

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

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

Answer 1

没有numpy的解决方案：

z=list(range(1,11))
offset=0
r=[]
p=[2,3]
for i in p*(len(z)//sum(p)):
    r.append(z[offset:offset+i])
    offset+=i

print(r)

的产率：

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