Numpy 数组的分割

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]]])]