五、Numpy——Array的分割

import  numpy as np
A = np.arange(12).reshape((3,4))
print(A)
print("==========纵向分割，等量分割============")
#分成两个矩阵 array，注意：分割的array数量必须可以整除A的列数
print(np.split(A,2,axis=1))#axis=1代表按列（纵向）分割
# [array([[0, 1],[4, 5],[8, 9]]),array([[ 2,  3],[ 6,  7],[10, 11]])]
print(np.hsplit(A,2))
# [array([[0, 1],[4, 5],[8, 9]]),array([[ 2,  3],[ 6,  7],[10, 11]])]
print("==========横向分割，等量分割============")
#按行分割，将矩阵分成三个array，注意：分割的array数量必须可以整除A的行数
print(np.split(A,3,axis=0))#axis=0代表按行（横向）分割
# [array([[0, 1, 2, 3]]), array([[4, 5, 6, 7]]), array([[ 8,  9, 10, 11]])]
print(np.vsplit(A,3))
#[array([[0, 1, 2, 3]]), array([[4, 5, 6, 7]]), array([[ 8,  9, 10, 11]])]
print("==========纵向分割，不等量分割============")
#分成三个矩阵 array，第一个array占两列，剩余列等量分割
print(np.array_split(A,3,axis=1))
# [array([[0, 1],[4, 5],[8, 9]]),array([[ 2],[ 6],[10]]),array([[ 3],[ 7],[11]])]