numpy argsort 의 의미와 사용법 정리

코드를 리뷰하다가 np.argsort 를 보고 정렬은 정렬인데 어떤식으로 정렬되는 건지 궁금해서 정리해보았다. 

 

각 원소의 정렬 인덱스 값을 반환한다.

numpy.argsort(a, axis=-1, kind=None, order=None)

b = [0,1,2,3,10,9,8]

 

x=np.argsort(b)

print(x)

결과값 

[0 1 2 3 6 5 4]

x2 = np.argsort(b)[::-1]. #내림차순으로 정렬

결과값

[4 5 6 3 2 1 0]

 

a = [[0,1,2,3],[1,2,3]]

print(np.argsort([len(aa) for aa in a]))

[1 0]

 

x = np.array([[0, 3], [2, 2]])

np.argsort(x, axis=0)

array([[0, 1],
       [1, 0]])

np.argsort(x, axis=1)

array([[0, 1],
       [0, 1]])