Numpy.sort, argsort

넘파이에서 행렬을 정렬하는 대표적인 함수로

np.sort( ) 와 ndarray.sort( ) 에 대해 알아보겠다.

 

행렬 정렬은 np.sort( )와 같이 넘파이에서

sort( ) 를 호출하는 방식과 ndarray.sort( )와 같이

행렬 자체에서 sort( )를 호출하는 방식이 있다.

 

두 방식의 차이는

np.sort( ) : 원본의 변경 없이 ( 원본 행렬 유지 ) 정렬된 배열의 복사본이 생성

ndarray.sort( ) : 원본 배열의 값이 정렬한 형태로 변경

 

import numpy as np

original_array = np.array([ 3, 1, 9, 5])

print('원본 행렬:', original_array)

#np.sort( )로 정렬

sort_array1 = np.sort(original_array)

print('np.sort() 호출 후 반환된 정렬 행렬:', sort_array1)

print('np.srot( ) 호출 후 원본 행렬 :', original_array)

#ndarray.sort( )로 정렬

sort_array2 = original_array.sort()

print('original_array.sort( ) 호출 후 반환된 행렬 : ', sort_array2)

print('original_array.sort( ) 호출 후 원본 행렬 : ', original_array)

 

[output]

원본 행렬: [3 1 9 5]

np.sort() 호출 후 반환된 정렬 행렬: [1 3 5 9]

np.srot( ) 호출 후 원본 행렬 : [3 1 9 5]

original_array.sort( ) 호출 후 반환된 행렬 : None

original_array.sort( ) 호출 후 원본 행렬 : [1 3 5 9]

 

 

원본 행렬 [3,1,9,5]에대해서

np.sort( ) 는 원본 행렬을 변경하지 않고

정렬된 형태로 반환했고

ndarray.sort( )는 원본 행렬 자체를 정렬한 값으로 변환했음을 알 수 있다.

np.sort( ) 와 ndarray.sort( ) 모두 오름차순으로 행렬 정렬을 디폴트로 하고있다.

내름차순으로 정렬하기 위해서는 np.sort( )[::-1] 을 사용한다.

 

sort_array1_desc = np.sort(original_array)[::-1]

print('내림차순으로 정렬:', sort_array1_desc)

[output]

내림차순으로 정렬: [9 5 3 1]

 

2차 행렬 이상일 경우 axis 축 값 설정을 통해서

로우 방향 또는 컬럼 방향으로 정렬을 할 수 있다.

 

array2d = np.array([[9, 15],

[10, 2]])

sort_array2d_axis0 = np.sort(array2d, axis=0)

print('로우 방향으로 정렬 : \n', sort_array2d_axis0)

sort_array2d_axis1 = np.sort(array2d, axis=1)

print('컬럼 방향으로 정렬 : \n', sort_array2d_axis1)

[output]

로우 방향으로 정렬 :

[[ 9 2]

[10 15]]

컬럼 방향으로 정렬 :

[[ 9 15]

[ 2 10]]

 

 

np.argsort( ) 는 정렬된 행렬의 원본 인덱스를 ndarray 형으로 반환한다.

 

original_array = np.array([3, 1, 9, 5])

sort_indices = np.argsort(original_array)

print(type(sort_indices))

print('행렬 정렬 시 원본 행렬의 인덱스 :', sort_indices)

[output]

<class 'numpy.ndarray'>

행렬 정렬 시 원본 행렬의 인덱스 : [1 0 3 2]

 

original_array = np.array([3, 1, 9, 5])

sort_indices_desc = np.argsort(original_array)[::-1]

print('행렬 내림차순 시 원본 행렬의 인덱스 :', sort_indices_desc)

[output]

행렬 내림차순 시 원본 행렬의 인덱스 : [2 3 0 1]

 

Reference

1.https://www.w3schools.com/python/numpy/numpy_array_sort.asp

NumPy Sorting Arrays

2. 파이썬 머신러닝 완벽가이드 위키북스 데이터 사이언스 시리즈