使用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")