python - 将元组/数组/列表解包为 Numpy 数组的索引

Question

我希望能够做到

>>> A = numpy.array(((1,2),(3,4)))
>>> idx = (0,0)
>>> A[*idx]

得到

1

但是这不是有效的语法。有没有办法在不明确写出的情况下做到这一点

>>> A[idx[0], idx[1]]

?

编辑:感谢您的回复。在我的程序中，我使用 Numpy 数组而不是元组进行索引并得到奇怪的结果。按照 Alok 的建议转换为元组就可以了。

Answer 1

这比你想象的要容易:

>>> import numpy
>>> A = numpy.array(((1,2),(3,4)))
>>> idx = (0,0)
>>> A[idx]
1