python--np.split()函数示例

我发现网上没有多少关于这个numpy.split函数的例子，我只好自己写一个。

沿不同轴进行分割，我们直接看例子：

import  numpy as np
A = np.arange(16).reshape(4,4)
print('A:',A)
print('沿1轴分割为2组：\n',np.split(A,2,axis = 1))
print('沿0轴分割为2组：\n',np.split(A,2,axis = 0))

输出为（A）：

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]
 [12 13 14 15]]

沿1轴分割为2组：

[array([[ 0,  1],
       [ 4,  5],
       [ 8,  9],
       [12, 13]]), array([[ 2,  3],
       [ 6,  7],
       [10, 11],
       [14, 15]])]

沿0轴分割为2组：

[array([[0, 1, 2, 3],
       [4, 5, 6, 7]]), array([[ 8,  9, 10, 11],
       [12, 13, 14, 15]])]