Replace numpy array with order of each value

Question

Hello I'm trying to do the following.

I have a matrix

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

I would like to transform the matrix so each cell new_a[i,j] contains the order (descending order) of the a[i,j] along the axis=1 So the result would be

a_new = np.array([[2,0,1], [2,1,0]])

I tried this (it works)

argsorts = np.argsort(a, axis=1) 

for i in a.shape[0]:
    a[i, argsorts[i]] = range(a.shape[1])[::-1]

I would like to change the for loop by something else Thank you

imdevskp · Accepted Answer · 2021-04-30 15:43:11Z

0

Try

>>> 2-a.argsort()
[[2 0 1]
 [2 1 0]]

answered Apr 30, 2021 at 15:43

imdevskp

2,2532 gold badges13 silver badges23 bronze badges

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

Collectives™ on Stack Overflow

Replace numpy array with order of each value

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