mapping two numpy arrays

Question

I have two numpy arrays A and B.

A = np.array ([[ 1  3] [ 2  3]  [ 2  1] ])

B = np.array([(1, 'Alpha'), (2, 'Beta'), (3, 'Gamma')]
 

How can I map A with B in order to get something like:

result = np.array ([[ 'Alpha'  'Gamma'] [ 'Beta'  'Gamma']  ['Beta'  'Alpha'] ])
 

I have tried this : map(B['f1'],A)  but I am getting TypeError: 'numpy.ndarray'  object is not callable. How can I solve this issue?

Answer 1

Here's a NumPythonic vectorized approach -

B[:,1][(A == B[:,0].astype(int)[:,None,None]).argmax(0)]
Sample run on a generic case -

In [118]: A
Out[118]:
array([[4, 3],
       [2, 3],
       [2, 4]])

In [119]: B
Out[119]:
array([['3', 'Alpha'],
       ['4', 'Beta'],
       ['2', 'Gamma']],
      dtype='|S5')

In [120]: B[:,1][(A == B[:,0].astype(int)[:,None,None]).argmax(0)]
Out[120]:
array([['Beta', 'Alpha'],
       ['Gamma', 'Alpha'],
       ['Gamma', 'Beta']],
      dtype='|S5')