import numpy as np
A = np.arange(12).reshape((3,4))
print(A)

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]

用split進行分割

split()包含三個參數，array的名稱、分割成幾份以及按哪個方向分割（axis = 0或1）

1. 上下分割 axis = 0

print(np.split(A,3,axis = 0))

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

2. 左右分割 axis = 1

print(np.split(A,2,axis = 1))

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

3. 用array_split進行不等量的分割

print(np.array_split(A,3,axis = 1))

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

用vsplit和hsplit進行分割

vsplit和hsplit包含兩個參數，array的名稱和分割成幾份

1. vertical split 上下分割

print(np.vsplit(A,3))

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

2. horizontal split 左右分割

print(np.hsplit(A,2))

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

Numpy學習教程來源請戳這里