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

【问题标题】：Numpy get values at indices given in array formNumpy 在以数组形式给出的索引处获取值
【发布时间】：2020-09-30 12:02:52
【问题描述】：

我想获取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 循环或列表推导？

【问题讨论】：

标签： python numpy numpy-ndarray numpy-slicing

【解决方案1】：

你可以使用np.take_along_axis:

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

【讨论】：