[파이썬] numpy array 정렬

1차원배열 정렬

• 좌에서 우로 정렬: np.sort(x)

• 거꾸로 정렬: np.sort(x)[::-1] , x[np.argsort(-x)]

  2차원배열 정렬

  • 열 축 기준(좌에서 우로)으로 정렬 : np.sort(x, axis=1)

  • 행 축 기준(위에서 아래로)으로 정렬: np.sort(x, axis=0)

  • 행 축 기준(위에서 아래로)으로 거꾸로 정렬: np.sort(x, axis=0)[::-1]

예시

1. np.sort(x)

   x = np.array([3,5,1]) --- 정렬 후 ---> array([1,3,5])

2. np.sort(x)[::-1] , x[np.argsort(-x)]

   x = np.array([3,5,1]) --- 정렬 후 ---> array([5,3,1])

3. np.sort(x, axis=1)

   x = np.array([3, 5, 1], array([1, 3, 5],

   [2, 6, 3], --- 정렬 후 ---> [2, 3, 6],

   [8, 3, 7]) [3, 7, 8])

4. np.sort(x, axis=0)

   x = np.array([3, 5, 1], array([2, 3, 1],

   [2, 6, 3], --- 정렬 후 ---> [3, 5, 3],

   [8, 3, 7]) [8, 6, 7])

5. np.sort(x, axis=0)[::-1]

   x = np.array([3, 5, 1], array([8, 6, 7],

   [2, 6, 3], --- 정렬 후 ---> [3, 5, 3],

   [8, 3, 7]) [2, 3, 1])

 

x.sort()와 np.sort(x)의 차이점

np.sort(x)는 원래 배열은 그대로, 정렬 결과의 복사본을 반환한다.

x.sort()는 배열 자체를 정렬한다.

참고

최초 발행 날짜: 2020-02-22