파이썬에는 sort와 sorted와 같은 정렬함수가 있습니다. 넘파이에도 똑같은 이름을 가진 함수가 있는데 오늘은 이것에 대해 알아보겠습니다.
넘파이에서는 sort 함수가 두 가지 종류가 있는데 먼저 원본을 보존하는 np.sort( )에 대해 알아보겠습니다.
import numpy as np
array1 = np.array([1, 10, 4, 7])
print(array1)
array2 = np.sort(array1)
print(array2)
print(array1)
[ 1 10 4 7]
[ 1 4 7 10]
[ 1 10 4 7]
정상적으로 정렬이 된 array2와 원본인 array1이 출력되는 것을 보실 수 있습니다.
이번에는 원본을 변경하는 ndarray.sort( )에 대해 알아보겠습니다.
array1 = np.array([1, 10, 4, 7])
print(array1)
array2 = array1.sort()
print(array2)
print(array1)
[ 1 10 4 7]
None
[ 1 4 7 10]
원본이 변경되어서 기존에 array1 = [1, 10, 4, 7]
로 저장하였던 array1의 값이 None으로 나오게 됩니다.
저번에 제가 앞으로 axis를 파라미터로 가지게 될 함수들이 많을 것이라고 했는데
sort( )가 아주 좋은 예시입니다.
만약 axis파라미터를 0으로 설정하면 row 방향으로 정렬이 되고 1로 설정하면 col 방향으로 정렬이 됩니다.
간단히 말해서 '[::-1]' 이것을 sort( )뒤에 붙여주면 됩니다.
이건 직접 코드를 작성해서 해보시길 바랍니다.
다음 포스팅에서도 정렬과 관련된 것을 설명할 예정이니 많은 관심 부탁드립니다.