python将一组数分成每3个一组

a = [1,2,3,4,5,6,7,8,9,10,11]
step = 3
b = [a[i:i+step] for i in range(0,len(a),step)]
print(b)

结果如下：

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

更新…
如果上面的step不是一个常数，比如

表示a的前三个分一组，接着的6个一组，再接着的9个分一组…，如果组数不够则结束

更新…2022/08/08

一个比较简单的方法是使用np.array_split()函数