두 개의 넘파이 배열을 같은 순서로 정렬하기

넘파이의 argsort 함수를 사용하여 첫 번째 배열을 오름순으로 정렬하는 인덱스를 얻은 다음, 두 번째 배열에 이 인덱스를 적용하여 첫 번째 배열과 동일한 순서로 정렬합니다. 

2023. 6. 27 최초작성

import numpy as np array1 = np.array([2, 1, 4, 3, 5]) array2 = np.array(['b', 'a', 'd', 'c', 'e']) # 첫번째 배열을 정렬하는 인덱스를 얻습니다. sort_indices = np.argsort(array1) # 인덱스를 사용하여 두 배열을 같은 순서로 정렬합니다. array1_sorted = array1[sort_indices] array2_sorted = array2[sort_indices] print(array1) print(array2) print() print(array1_sorted) print(array2_sorted)

실행결과입니다. 두 배열 간의  순서를 유지한채 첫번째 배열을 기준으로 오름차순으로 정렬되도록 합니다. 

 

[2 1 4 3 5]

['b' 'a' 'd' 'c' 'e']

 

[1 2 3 4 5]

['a' 'b' 'c' 'd' 'e']