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])
 
 

 

posted @ 2021-12-01 15:01  ArdenWang  阅读(99)  评论(0编辑  收藏  举报