拆分Numpy数组

hspilt() vspilt() dspilt() spilt()

np. hspilt()

a = np.arange(9).reshape(3,3)
print(np.hsplit(a,3)) 
# 沿着横轴方向分割 相当于np.split(a, 3，axis=1)

array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])

[array([[0],
       [3],
       [6]]), array([[1],
       [4],
       [7]]), array([[2],
       [5],
       [8]])]

np.vsplit(a,3)

np.vsplit(a,3)
#相当于 np.spilt(a,3,axis=0)
[array([[0, 1, 2]]), array([[3, 4, 5]]), array([[6, 7, 8]])]

np.dsplit() 深度分解数组

a = np.arange(27).reshape(3,3,3)
array([[[ 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]]])

np.dsplit(a,3)
[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]]])]