python numpy 分割

#python numpy 分割

import numpy as np
a=np.arange(12).reshape((3,4))
print(a)

print(np.split(a,2,axis=1))  #把行分成两份
print(np.vsplit(a,3))        #纵向分成三块
print(np.hsplit(a,2))         #横向分成两块

输出：

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