分区中的一维数组成不同大小的块依次用numpy的

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]]