我想使用大小为（N，）的数组索引维度为N的数组。例如，让我们考虑N为2的情况。
import numpy as np
foo = np.arange(9).reshape(3,3)
bar = np.array((2,1))
>>> foo
array([[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])
>>>bar
array([2, 1])
>>>foo[bar[0],bar[1]]
7
这很好。但是，使用这种方法，我将需要编写N次bar [i]，如果N高，这不是一个很好的解决方案。
以下命令没有给出我需要的结果：
>>>foo[bar]
array([[6, 7, 8],
[3, 4, 5]])
我怎样做才能以一种简洁的方式获得想要的结果？
我认为你可以bar变成元组：
foo[tuple(bar)]
# 7
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句