numpy的array分割

import numpy as np

a = np.arange(12).reshape((3, 4))
print("# 原始数组是:\n", a, end="\n\n")
# 使用np.split(a, 3, axis=0)进行分割数组，axis=0是按照行分割，axis=1是按照列分割
b = np.split(a, 3, axis=0)
print("# 按照行分割:\n", b, end="\n\n")
# 在分割的时候要确定能够整除
c = np.split(a, 2, axis=1)
print("# 按照列分割:\n", c, end="\n\n")

# 除不尽时也可以分割,前面的array会多出来一列，直到列用尽
d = np.array_split(a, 3, axis=1)
print("# 分为不等的array:\n", d, end="\n\n")

# 也可以用下面方法进行分割
e = np.vsplit(a, 3)  # 进行按行分割
print("# 进行按行分割：\n", e, end="\n\n")
e = np.hsplit(a, 2)  # 进行按列分割
print("# 进行按列分割：\n", e, end="\n\n")