a = np.arange(9).reshape(3,3)
b = 2 * a
01水平分割 hsplit 或者 split 参数 axis = 1 都可以实现
c = np.hsplit(a,3)
print(c)
#输出结果
[array([[0],
[3],
[6]]),
array([[1],
[4],
[7]]),
array([[2],
[5],
[8]])]
02垂直 分割 vsplit 或者 split函数axis=0都可以完成
d = np.vsplit(a,3)
print(d)
#输出结果
[array([[0, 1, 2]]),
array([[3, 4, 5]]),
array([[6, 7, 8]])]
03 深度分割 dsplit 按照深度方向分割数组
e = np.arange(27).reshape(3,3,3)
print(e)
f = np.dsplit(e,3)
print(f)
#输出结果
[[[ 0 1 2]
[ 3 4 5]
[ 6 7 8]]
[[ 9 10 11]
[12 13 14]
[15 16 17]]
[[18 19 20]
[21 22 23]
[24 25 26]]]
[array([[[ 0],
[ 3],
[ 6]],
[[ 9],
[12],
[15]],
[[18],
[21],
[24]]]), array([[[ 1],
[ 4],
[ 7]],
[[10],
[13],
[16]],
[[19],
[22],
[25]]]), array([[[ 2],
[ 5],
[ 8]],
[[11],
[14],
[17]],
[[20],
[23],
[26]]])]