np.slipt()函数的用法说明及举例

split(ary, indices_or_sections, axis=0)

功能：把一个数组从在第0维按顺序切分参数：
ary:要切分的数组；
indices_or_sections:如果是一个整数，就用该数平均切分，如果是一个数组，为沿轴切分的位置（左开右闭）；
axis：沿着哪个维度进行切向，默认为0，横向切分。为1时，纵向切分。

下面按照indices_or_sections参数为整数或数组分别举例。

indices_or_sections参数为整数

b=np.arange(30).reshape(10,3)
print(b)

a1 = np.split(b,2,axis=0)
a2= np.split(b,[2,6,8],axis=0)
print(a1)
print(a2)

[[ 0  1  2]
 [ 3  4  5]
 [ 6  7  8]
 [ 9 10 11]
 [12 13 14]
 [15 16 17]
 [18 19 20]
 [21 22 23]
 [24 25 26]
 [27 28 29]]

#整数结果，平分两份
[array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 6,  7,  8],
       [ 9, 10, 11],
       [12, 13, 14]]), array([[15, 16, 17],
       [18, 19, 20],
       [21, 22, 23],
       [24, 25, 26],
       [27, 28, 29]])]
#数组结果，按数组里表示的位置（按第几行）分割
[array([[0, 1, 2],
       [3, 4, 5]]), array([[ 6,  7,  8],
       [ 9, 10, 11],
       [12, 13, 14],
       [15, 16, 17]]), array([[18, 19, 20],
       [21, 22, 23]]), array([[24, 25, 26],
       [27, 28, 29]])]