numpy[:,1]
第a维坐标为b的数据
[:,:,...:b]-共a个
import numpy as np
# 3维
a = np.array([[[1,2],[3,4]]]) a[:,:,1] output: array([[2, 4]]) input:a[:,1,:] output:
array([[3, 4]])
# 2维
b = np.array([[1,2,3]]) b[:,1] output: array([2]) input:b[:,2] #第二维,坐标为2的数据 array([3])