Dada una matriz numpy autoindexada (no estoy seguro si este es el término correcto), por ejemplo:

a = np.array([3, 2, 0, 1])

Esto representa estopermutación (=> es una flecha):

0 => 3
1 => 2
2 => 0
3 => 1

Estoy tratando de hacer una matriz que represente la transformación inversa sin hacerlo "manualmente" en Python, es decir, quiero unpuro solución numpy El resultado que quiero en el caso anterior es:

array([2, 3, 1, 0])

Que es equivalente a

0 <= 3                0 => 2
1 <= 2       or       1 => 3
2 <= 0                2 => 1
3 <= 1                3 => 0

Parece tan simple, pero no puedo pensar en cómo hacerlo. He intentado googlear, pero no he encontrado nada relevante.