Problems sorting an 2D array

Question

I have a following 2D numpy array:

array([[1 0]
       [2 0]
       [4 0]
       [1 1]
       [2 1]
       [3 1]
       [4 2])

I want to sort the ID of first column with its value in second value, suck that I get back:

array([[1 0]
       [1 1]
       [2 0]
       [2 1]
       [3 1]
       [4 0]
       [4 2]])

I am getting O(n^2) complexity and want to improve it further.

Have you tried using numpy sort het.as.utexas.edu/HET/Software/Numpy/reference/generated/… ? — Drecker
– Drecker, Commented Sep 11, 2019 at 10:46
@Austin this is just 2 column in my data, if I show you my solution, I has to fix all question — CuCaRot
– CuCaRot, Commented Sep 11, 2019 at 10:49

Hu Xixi · Accepted Answer · 2019-09-11 11:49:58Z

1

A better way to sort a list of lists:

import numpy as np
a = np.array([[1, 0], [2 ,0], [4 ,0], [1 ,1], [2 ,1], [3 ,1], [4 ,2]])
s_a = np.asarray(sorted(a, key=lambda x: x[0]))
print(s_a)

Output:

[[1 0]
 [1 1]
 [2 0]
 [2 1]
 [3 1]
 [4 0]
 [4 2]]

answered Sep 11, 2019 at 11:49

Hu Xixi

2,1972 gold badges26 silver badges32 bronze badges

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

1 Comment

Hu Xixi Over a year ago

@Cuz.I'm.BatMan You're welcome, I'm glad to be helpful.

Shishir Naresh · Accepted Answer · 2019-09-11 10:52:00Z

1

Try the below code, Hope this will help:

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

np.sort(a.view('i8,i8'), order=['f0'], axis=0).view(np.int)

Ouput will be :

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

edited Sep 11, 2019 at 10:52

answered Sep 11, 2019 at 10:49

Shishir Naresh

7634 silver badges10 bronze badges

2 Comments

CuCaRot Over a year ago

I got this bug 'ValueError: When changing to a larger dtype, its size must be a divisor of the total size in bytes of the last axis of the array.'

Shishir Naresh Over a year ago

@Cuz.I'm.BatMan Please upvote and accept my answer, if it works for you.

Collectives™ on Stack Overflow

Problems sorting an 2D array

2 Answers 2

1 Comment

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related