desembalaje tuplas/arreglos/listas como índices para numpy matrices

Question

me gustaría ser capaz de hacer

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

y obtener

1 

sin embargo esto no es una sintaxis válida. ¿Hay alguna forma de hacerlo sin escribir explícitamente

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

?

EDIT: Gracias por las respuestas. En mi programa estaba indexando con una matriz Numpy en lugar de una tupla y obtenía resultados extraños. La conversión a una tupla como Alok sugiere el truco.

Answer 1

Es más fácil de lo que piensa:

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

Answer 2

Trate

A[tuple(idx)] 

A menos que tenga un caso de uso más complejo que no es tan simple como este ejemplo, lo anterior debe funcionar para todas las matrices.

Answer 3

Indexación de un objeto llama:

object.__getitem__(index) 

Al hacer A [1, 2], que es el equivalente de:

A.__getitem__((1, 2)) 

Así que cuando lo hace:

b = (1, 2) 

A[1, 2] == A[b] 
A[1, 2] == A[(1, 2)] 

Ambas declaraciones evaluarán a True.

Si le sucede al índice con una lista, que podría No índice de la misma, como [1, 2]! = (1, 2)

Answer 4

Sin desembalaje es necesario -cuando tiene una coma entre [ y ], está haciendo una tupla, no pasando argumentos. foo[bar, baz] es equivalente a foo[(bar, baz)]. Entonces, si tiene una tupla t = bar, baz, simplemente diga foo[t].