특정 열을 기준으로 NumPy 배열을 어떻게 정렬할 수 있나요?

열을 기준으로 NumPy 배열 정렬

특정 열을 기준으로 NumPy 배열을 정렬하는 것은 데이터 조작에 필요한 작업인 경우가 많습니다. 예를 들어 다음 배열을 고려해보세요.

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

두 번째 열의 값을 기준으로 a의 행 순서를 변경한다고 가정해 보겠습니다. 이를 달성하기 위해 NumPy에서 argsort() 함수를 활용할 수 있습니다.

a[:, 1].argsort()

위 표현식은 argsort()를 a의 두 번째 열에 적용하여 정렬된 값에 해당하는 인덱스 배열을 생성합니다. 구체적으로 다음을 반환합니다.

array([1, 0, 2])

이러한 인덱스는 오름차순으로 두 번째 열을 기준으로 정렬하기 위해 a의 행을 다시 정렬해야 하는 순서를 나타냅니다. 이를 배열 인덱싱과 결합하면 정렬된 배열을 얻을 수 있습니다.

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

이 표현식은 다음을 반환합니다.

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

원하는 대로 a의 행은 두 번째 열에 따라 정렬되었습니다. .

위 내용은 특정 열을 기준으로 NumPy 배열을 어떻게 정렬할 수 있나요?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!