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 循環或列表理解是否可能？
1 回答
牛魔王的故事
TA貢獻1830條經驗 獲得超3個贊
您可以使用np.take_along_axis：
np.take_along_axis(my_array,?indices,?axis=-1) array([[[1.1587323?], ????????[1.29215026], ????????[0.9784655?]]])
添加回答
舉報
0/150
提交
取消