Python NumPy에서 배열을 열별로 정렬하기, Sorting arrays in NumPy by column

질문

NumPy 배열을 n번째 열을 기준으로 정렬하는 방법은 무엇인가요?

예를 들어, 다음과 같이 주어진 경우:

a = array([[9, 2, 3],
           [4, 5, 6],
           [7, 0, 5]])

a의 행을 두 번째 열을 기준으로 정렬하여 다음과 같이 얻고 싶습니다:

array([[7, 0, 5],
       [9, 2, 3],
       [4, 5, 6]])

답변

두 번째 열을 기준으로 a를 정렬하려면:

a[a[:, 1].argsort()]