有时候需要将一个很长的列表按一定的顺序分成嵌套列表,用列表操作会有点小麻烦,需要先将列表切片然后在添加到不同的列表中,但是用numpy库中的数组来操作就很方便,那就是数组的形态重塑功能,话不多说,代码如下:
import numpy as np
number = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
arr = np.array(number).reshape(5, 2) #将列表分成5个嵌套列表,每个列表长度为2
print(arr)
结果如下:
[[ 1 2]
[ 3 4]
[ 5 6]
[ 7 8]
[ 9 10]]
这样就能很快的将列表划分了,代码也很简单 ,但是要注意哦,使用reshape重塑数组形态的时候,原数组里的元素个数必须能够被划分完