How to substitute values from array to values in list in Python?

Question

I have an array (of numpy.ndarray type) like

arr=array([[1, 5, 1],
          [4, 2, 0]])

and a list of values:

values=['father','mother','sister','brother','aunt','uncle']

And I'd like to substitute numbers in array arr with items from list values using array's items as indices of list values:arr[0,0]=values[arr[0,0]]

Here an example of what I'd like to have

arr=array([['mother', 'uncle', 'mother'],
           ['aunt', 'sister', 'father']])

Is there any elegant pythonic way to do this?

Thanks for helping in advance =)

Kasravnd · Accepted Answer · 2016-03-25 09:23:19Z

2

You can convert the values to a numpy array then use a simple indexing:

>>> values = np.array(values)
>>> 
>>> values[arr]
array([['mother', 'uncle', 'mother'],
       ['aunt', 'sister', 'father']], 
      dtype='|S7')

Read more about indexing: http://docs.scipy.org/doc/numpy-1.10.0/reference/arrays.indexing.html

answered Mar 25, 2016 at 9:23

Kasravnd

108k19 gold badges167 silver badges195 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

B. M. · Accepted Answer · 2016-03-25 13:13:17Z

1

use the numpy take function :

In [64]: np.take(values,arr)
Out[64]: 
array([['mother', 'uncle', 'mother'],
       ['aunt', 'sister', 'father']], 
      dtype='<U7')

The conversion is then automatic.

answered Mar 25, 2016 at 13:13

B. M.

18.7k2 gold badges40 silver badges56 bronze badges

Collectives™ on Stack Overflow

How to substitute values from array to values in list in Python?

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related