Numpy 获取以数组形式给出的索引处的值

Question

indices我想获取 的值my_array。indices = np.array([[[0],         [1],         [0]]]) my_array = np.array([[[1.1587323 , 1.75406635],         [1.05464125, 1.29215026],         [0.9784655 , 1.16957462]]])我应该得到以下输出：output: array([[[1.1587323], [1.29215026], [0.9784655]]])没有 for 循环或列表理解是否可能？

Answer 1

您可以使用np.take_along_axis：

np.take_along_axis(my_array, indices, axis=-1)
array([[[1.1587323 ],
        [1.29215026],
        [0.9784655 ]]])