특정 열을 기준으로 NumPy 배열을 정렬하는 방법은 무엇입니까?

특정 열을 기준으로 NumPy 배열 정렬

이 쿼리는 지정된 n번째 열에 따라 NumPy 배열을 정렬해야 하는 필요성을 해결합니다. 설명을 위해 배열 'a'로 작업해 보겠습니다.

import numpy as np
a = np.array([[9, 2, 3],
           [4, 5, 6],
           [7, 0, 5]])

우리의 목표는 두 번째 열을 기준으로 행렬 'a'의 행을 정렬하여 다음과 같은 결과를 얻는 것입니다.

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

이를 달성하기 위해 인덱스를 기반으로 NumPy 배열을 분할하고 argsort 기능을 활용하는 기능을 활용할 수 있습니다. 아래 코드는 해결 방법을 보여줍니다.

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

이 코드 분석:

1. a[:, 1]: 이는 '의 두 번째 열을 선택합니다. a'.
2. .argsort(): 이는 다음의 배열을 생성합니다. 두 번째 열에 있는 요소의 정렬된 순서를 나타내는 인덱스입니다.
3. a[a[:, 1].argsort()]: 이는 'a' 행을 다음에 따라 다시 정렬합니다. 색인을 정렬하여 원하는 결과를 얻습니다.

위 내용은 특정 열을 기준으로 NumPy 배열을 정렬하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!