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

NumPy에서 특정 열을 기준으로 다차원 배열 정렬

NumPy는 지정된 열을 기준으로 다차원 배열을 정렬하는 등 강력한 배열 조작 기능을 제공합니다. 이 기능은 데이터를 구성하고 분석하는 데 특히 유용합니다. NumPy에서 열 기반 정렬을 수행하는 방법을 살펴보겠습니다.

여러 열이 있는 NumPy 배열 a가 있다고 가정합니다.

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

두 번째 열을 기준으로 a의 행을 정렬하려면 다음을 수행합니다. argsort() 함수를 활용할 수 있습니다. 배열을 가져와서 배열을 정렬할 인덱스를 반환합니다. 이러한 정렬된 인덱스로 a를 인덱싱하면 원하는 정렬된 배열을 얻을 수 있습니다.

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

이 작업으로 인해 다음과 같은 정렬된 배열이 생성됩니다.

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

구문 a[:, 1 ] 배열 a에서 두 번째 열(인덱스 1)을 선택합니다. 이 열에 적용된 argsort()는 정렬된 인덱스를 생성한 다음 행을 정렬하는 데 사용됩니다.

이 기술은 매우 효율적이고 다용도이므로 다차원 배열에서 원하는 열을 기준으로 정렬할 수 있습니다. 데이터에서 의미 있는 통찰력을 구성하고 추출하는 편리하고 강력한 방법을 제공합니다.

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