Numpy 배열을 열별로 정렬하는 방법(예제 포함)

다음 방법을 사용하여 NumPy 배열의 행을 열 값으로 정렬할 수 있습니다.

방법 1: 열 값을 기준으로 오름차순으로 정렬

 x_sorted_asc = x[x[:, 1]. argsort ()]

방법 2: 열 값을 내림차순으로 정렬

 x_sorted_desc = x[x[:, 1]. argsort ()[::-1]]

다음 예에서는 각 방법을 실제로 사용하는 방법을 보여줍니다.

예시 1: 열 값을 기준으로 Numpy 배열을 오름차순으로 정렬

다음과 같은 NumPy 배열이 있다고 가정합니다.

 import numpy as np

#create array
x = np. array ([14, 12, 8, 10, 5, 7, 11, 9, 2]). reshape (3,3)

#view array
print (x)

[[14 12 8]
 [10 5 7]
 [11 9 2]]

다음 코드를 사용하여 NumPy 테이블의 행을 두 번째 열의 값을 기준으로 오름차순으로 정렬할 수 있습니다.

 #define new matrix with rows sorted in ascending order by values in second column
x_sorted_asc = x[x[:, 1]. argsort ()]

#view sorted matrix
print (x_sorted_asc)

[[10 5 7]
 [11 9 2]
 [14 12 8]]

이제 행은 두 번째 열의 값을 기준으로 오름차순(가장 작은 것부터 가장 큰 것까지)으로 정렬됩니다.

예 2: 열 값을 내림차순으로 Numpy 배열 정렬

다음과 같은 NumPy 배열이 있다고 가정합니다.

 import numpy as np

#create array
x = np. array ([14, 12, 8, 10, 5, 7, 11, 9, 2]). reshape (3,3)

#view array
print (x)

[[14 12 8]
 [10 5 7]
 [11 9 2]]

다음 코드를 사용하여 NumPy 테이블의 행을 두 번째 열의 값을 기준으로 내림차순으로 정렬할 수 있습니다.

 #define new matrix with rows sorted in descending order by values in second column
x_sorted_desc = x[x[:, 1]. argsort ()[::-1]]

#view sorted matrix
print (x_sorted_desc)

[[14 12 8]
 [11 9 2]
 [10 5 7]]

이제 행은 두 번째 열의 값을 기준으로 내림차순(최대에서 최소)으로 정렬됩니다.

추가 리소스

다음 튜토리얼에서는 Python에서 다른 일반적인 작업을 수행하는 방법을 설명합니다.

NumPy 배열에서 값 인덱스를 찾는 방법
NumPy 배열에서 특정 열을 얻는 방법
NumPy 배열에 열을 추가하는 방법