numpy.argsort¶

numpy.argsort 함수는 어레이를 정렬하는 인덱스의 어레이를 반환합니다.

예제1¶

import numpy as np

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

print(np.argsort(a))

[1 3 0 2]

np.argsort(a)는 인덱스의 어레이 [1 3 0 2]를 반환합니다.

import numpy as np

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

print(a[sort_ind])

[1 2 3 4]

a[sort_ind]와 같이 정렬 인덱스를 사용해서 어레이를 정렬할 수 있습니다.