Numpy: sort by key function

Question

Is there a way to sort the rows of a numpy ndarray using a key (or comparator) function, without resorting to converting to a python list?

In particular, I need to sort according to this function:

c1,c2= 4,7
lambda row: c1*(row[1]/c2)+row[0]

I realise one possible solution would be to generate a vector with the key value of each row, but how would one sort according to it? Should one seek to convert such vector into a index vector somehow?

order= c1*(matrix[:,1]/c2)+matrix[:,0]
indexes= order_to_index( order )
return matrix[ indexes ]

Is this realistic?

@wiso indeed! it never occurred to me that this is the same problem as giving the indexes that order the key value vector and sorting by it! — loopbackbee
– loopbackbee, Commented Oct 13, 2012 at 23:35
just try: np.argsort([3,6,2,2,8]) #-> array([2, 3, 0, 1, 4]). It doesn't sort your original input. Usually numpy functions don't have side effects — Ruggero Turra
– Ruggero Turra, Commented Oct 13, 2012 at 23:38
@wiso indeed, just did, I edited my last post in the meantime, sorry. Thank you! Will you consider adding an answer? — loopbackbee
– loopbackbee, Commented Oct 13, 2012 at 23:40

Ruggero Turra · Accepted Answer · 2012-10-13 23:55:15Z

7

your approach is right, it is similar to the Schwartzian transform or Decorate-Sort-Undecorate (DSU) idiom

As I said you can use the numpy function np.argsort. It does the work of your order_to_index.

edited Oct 13, 2012 at 23:55

answered Oct 13, 2012 at 23:42

Ruggero Turra

17.8k21 gold badges92 silver badges148 bronze badges

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

Comments

Dillon Bowen · Accepted Answer · 2021-04-29 17:07:59Z

6

For a more explicit answer, suppose we have an array x and want to sort the rows according to some function func which takes a row of x and outputs a scalar.

x[np.apply_along_axis(func, axis=1, arr=x).argsort()]

For this example

c1, c2 = 4, 7
x = np.array([
    [0, 1],
    [2, 3],
    [4, -5]
])
x[np.apply_along_axis(lambda row: c1 * / c2 * row[1] + row[0], 1, x).argsort()]

Out:

array([[ 0,  1],
       [ 4, -5],
       [ 2,  3]])

In this case, np.apply_along_axis isn't even necessary.

x[(c1 / c2 * x[:,1] + x[:,0]).argsort()]

Out:

array([[ 0,  1],
       [ 4, -5],
       [ 2,  3]])

answered Apr 29, 2021 at 17:07

Dillon Bowen

3961 gold badge4 silver badges9 bronze badges

Collectives™ on Stack Overflow

Numpy: sort by key function

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related