NumPy 어레이 정렬 np.argsort()

np.argsort() 정리하기

-> 오름차순으로 정렬된 값의 인덱스 반환한다.

import numpy as np

a = np.array([1.5, 0.2, 4.2, 2.5])
print(np.argsort(a)) # [1 0 3 2]
print(a.argsort()) # [1 0 3 2]

정렬한 값 출력하기 (오름차순)

b = a.argsort()
print(a) # [1.5 0.2 4.2 2.5]

print(a[b]) # [0.2 1.5 2.5 4.2]
print(a[a.argsort()] # [0.2 1.5 2.5 4.2]

내림차순 정렬하기

#내림차순으로 정렬하기
print(a[b[::-1]]) # [4.2 2.5 1.5 0.2]

print(a[b][::-1]) # [4.2 2.5 1.5 0.2]

가장 큰 값 3개 꺼내기

# 가장 큰 값 3개 꺼내기
print(a[b][-3:])

# 가장 큰 값 n개 꺼내기
print(a[b][-n:])