numpy_分割

import numpy as np

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

# 第一种分割 split---进行均匀分割
print(np.split(a, 3, axis=0))  # 横向分割3份
print(np.split(a, 4, axis=1))  # 纵向分割4份
print(np.split(a, 4, axis=0))  # 报错，split分割不均匀

# 第二种分割 array_split---进行不等分割
print(np.array_split(a, 3, axis=1))  # 成功，array_split可以进行不等分割

# 第三种分割 vssplit---指定横向分割, hsplit---指定纵向分割
print(np.vsplit(a, 3))  # 横向分割成3份
print(np.hsplit(a, 4))  # 纵向分割成4份