How can I assign strings to different elements in an array, sort the array, then display the strings based on the sort in Python/Numpy?

Question

I have an array P = np.array([2,3,1]) and I want to assign three strings to each element respectively. So,:

"str1", "str2", "str3" = P

and after sorting the array:

In[]: P = -np.sort(-P)
Out[]: [3,2,1]

I want to then be able to display the strings based on this sort, as:

Out[]: "str2","str1","str3",

Tried assigning variable names to the elements but it won't display on output as intended. Tried defining an array of objects with the strings as elements but have trouble assigning them to the numerical values of P.

There is no such thing as "assigning strings", if you are trying to create dynamic variables know that this is very bad practice (and often useless). What are you really trying to achieve? — mozway
– mozway, Commented Jan 5, 2023 at 17:43

Mahdi F. · Accepted Answer · 2023-01-05 17:43:09Z

1

You can use numpy.argsort.

import numpy as np
P = np.array([2,3,1])
S = np.array(["str1", "str2", "str3"])

sort_idx = np.argsort(-P)
print(S[sort_idx])
# ['str2' 'str1' 'str3']

answered Jan 5, 2023 at 17:43

Mahdi F.

24.1k5 gold badges25 silver badges32 bronze badges

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

Collectives™ on Stack Overflow

How can I assign strings to different elements in an array, sort the array, then display the strings based on the sort in Python/Numpy?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related