【Python】split()

import numpy as np

b= np.arange(12).reshape(3,4)
print(b)
print(np.split(b,2,axis=1))
print(np.split(b,3,axis=0))


#axis=1表示沿着每一行下去分割(一共三行)，分成2列，故为
#[array([[0, 1],
#       [4, 5],
#       [8, 9]]), array([[ 2,  3],
#       [ 6,  7],
#       [10, 11]])]
#axis=0表示沿着每一列下去分割(一共四列)，分成三行，故为
#[array([[0, 1, 2, 3]]), array([[4, 5, 6, 7]]), array([[ 8,  9, 10, 11]])]

很多时候，如果不明白的话，可以动手画一画，一下就明白了