Python中的numpy数组的分割

使用hsplit、vsplit、dsplit进行数组的分割

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

# 水平分割
print("水平分割")
b=np.hsplit(a,3)
print(b)
print("----------------\n")
b1=np.split(a,3,axis=1)
print(b1)
print("-----------------------------------\n")

# 垂直分割
print("垂直分割")
c=np.vsplit(a,3)
print(c)
print("----------------\n")
c1=np.split(a,3,axis=0)
print(c1)
print("-----------------------------------\n")

# 深度分割
print("深度分割")
d=np.arange(27).reshape(3,3,3)
d1=np.dsplit(d,3)
print(d1)
print("-----------------------------------\n")