如何用 python 中的列表字符串替换 numpy 数组的数字

Question

I have a numpy array and want to replace its values with some strings. I checked this solution but could not solve my issue. My strings are in a list:

names=['base', 'middle', 'top']

my list has three strings and it means I have also three numbers in my array:

import numpy as np
numbers=np.array([[3., 1., 2., 2.], [1., 2., 3., 1.]])

To map names with numbers I have this logic: from the first name until the last, numbers decrease. I mean first name matches the highest number ( 'base': 3. ), next name the second highest ( 'middle': 2. ) and last name equals the lowest ( 'top': 1. ). Finally I want to have my array as:

subs_arr= np.array([['base', 'top', 'middle', 'middle'], ['top', 'middle', 'base', 'top']])

Is ther any way to do it automatically (without the need for mapping names with numbers) in python? I do appreciate any help in advance.

Answer 1

How about smth like this:

import numpy as np
numbers = np.array([[3., 1., 2., 2.], [1., 2., 3., 1.]]).astype(int)-1
names = np.array(['top', 'middle', 'base'])

subs_arr = names[numbers]

output:

subs_arr = 
[['base' 'top' 'middle' 'middle']
 ['top' 'middle' 'base' 'top']]