Traduire chaque élément d'un tableau numpy en fonction de la clé.

Question

J'essaie de traduire chaque élément d'un numpy.array en fonction d'une clé donnée :

Par exemple :

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

my_dict = {1:23, 2:34, 3:36, 4:45}

Je veux avoir :

array([[ 23.,  34.,  36.],
       [ 36.,  34.,  45.]])

Je peux voir comment le faire avec une boucle :

def loop_translate(a, my_dict):
    new_a = np.empty(a.shape)
    for i,row in enumerate(a):
        new_a[i,:] = map(my_dict.get, row)
    return new_a

Existe-t-il une méthode plus efficace et/ou purement numpy ?

Edita:

Je l'ai chronométré, et np.vectorize La méthode proposée par DSM est considérablement plus rapide pour les grandes matrices :

In [13]: def loop_translate(a, my_dict):
   ....:     new_a = np.empty(a.shape)
   ....:     for i,row in enumerate(a):
   ....:         new_a[i,:] = map(my_dict.get, row)
   ....:     return new_a
   ....: 

In [14]: def vec_translate(a, my_dict):    
   ....:     return np.vectorize(my_dict.__getitem__)(a)
   ....: 

In [15]: a = np.random.randint(1,5, (4,5))

In [16]: a
Out[16]: 
array([[2, 4, 3, 1, 1],
       [2, 4, 3, 2, 4],
       [4, 2, 1, 3, 1],
       [2, 4, 3, 4, 1]])

In [17]: %timeit loop_translate(a, my_dict)
10000 loops, best of 3: 77.9 us per loop

In [18]: %timeit vec_translate(a, my_dict)
10000 loops, best of 3: 70.5 us per loop

In [19]: a = np.random.randint(1, 5, (500,500))

In [20]: %timeit loop_translate(a, my_dict)
1 loops, best of 3: 298 ms per loop

In [21]: %timeit vec_translate(a, my_dict)
10 loops, best of 3: 37.6 ms per loop

In [22]:  %timeit loop_translate(a, my_dict)

Answer 1

Si tu ne veux pas vraiment doivent utiliser le dictionnaire comme tableau de substitution, la solution simple serait (pour votre exemple) :

a = numpy.array([your array])
my_dict = numpy.array([0, 23, 34, 36, 45])     # your dictionary as array

def Sub (myarr, table) :
    return table[myarr] 

values = Sub(a, my_dict)

Cela ne fonctionnera bien sûr que si les index de d couvrir toutes les valeurs possibles de votre a en d'autres termes, seulement pour a avec des entiers signés.