>백엔드 개발 >파이썬 튜토리얼 >특정 열을 기준으로 NumPy 배열의 행을 정렬하는 방법은 무엇입니까?

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

DDD
DDD원래의
2024-11-28 21:27:18841검색

How to Sort a NumPy Array's Rows by a Specific Column?

NumPy에서 열을 기준으로 배열 정렬

열을 기준으로 배열을 정렬하는 것은 데이터 분석 및 조작에서 일반적인 작업입니다. NumPy는 이 작업을 수행하는 효율적인 방법을 제공합니다.

질문:

여러 열이 있는 NumPy 배열이 있는 경우 특정 열을 기준으로 행을 어떻게 정렬할 수 있습니까? 열?

예:

다음 배열 a를 고려해보세요.

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

a의 행을 두 번째 열( 열 1, 인덱스는 0부터)부터 획득:

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

답변:

두 번째 열을 기준으로 a를 정렬하려면 다음 코드를 사용합니다.

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

이 표현식은 다음을 달성합니다. 원하는 결과는 다음과 같습니다:

  1. a[:, 1]이 두 번째 열을 선택합니다. of a.
  2. argsort()는 열을 오름차순으로 정렬하는 인덱스를 반환합니다.
  3. 이 인덱스로 a를 인덱싱하면 그에 따라 a의 행이 정렬됩니다.

따라서 이 접근 방식은 지정된 열을 기준으로 NumPy 배열을 정렬하는 편리하고 효율적인 방법을 제공합니다.

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

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.