Indexation du tableau entre deux tableaux

J'ai une matrice numpue et des valeurs de recherche correspondantes. Dans toutes les colonnes de données, je devrais obtenir une position maximale de la recherche correspondante un convertissant cette position en résultat des données. P>

Je dois faire comme indiqué sur la photo. P>

image parle mieux que ma langue. P>

data = data.transpose(1,0)

lookup = lookup.transpose(1,0)

results = []       
for d in data:   
   matches = lookup[d].diagonal()
   print (matches)
   idx = np.argmax(matches, axis=0)
   res = d[idx]
   results.append(res)

print ()

print (results)