Python : numpy切分

""“切分”""
import numpy as np
n1 = np.random.randint(1,100,size=(5,5))
n3 = np.split(n1,(1,4),axis=0) # axis=0,行切分，1列切分
#print(n3)

运行结果：

[array([[88, 18, 14, 94, 99]]),

array([[50, 19, 34, 72, 15],

[ 7, 8, 49, 45, 1],

[98, 19, 29, 85, 57]]),

array([[96, 27, 30, 27, 43]])]

n4 = np.split(n1,(1,4),axis=1)
#print(n4)

运行结果：

[array([[44],

[94],

[78],

[62],

[82]]), array([[ 9, 71, 32],

[54, 15, 23],