NumPy 배열에서 여러 최대값의 인덱스를 찾는 방법은 무엇입니까?

NumPy 배열에서 여러 최대값의 인덱스 검색

NumPy 배열은 최대 요소의 인덱스를 찾는 np.argmax 함수를 제공합니다. . 그러나 N 최대값의 인덱스가 필요한 경우 다음 솔루션을 고려하십시오.

최근 NumPy 버전:

NumPy 버전 1.8 이상의 경우 argpartition 함수는 다음을 제공합니다. 효율적인 방법:

import numpy as np

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

# Get top 4 indices
n_max = 4
ind = np.argpartition(a, -n_max)[-n_max:]

# Retrieve top 4 values
top_max = a[ind]

이전 NumPy 버전:

NumPy 1.8 이전에는 다음과 같이 argsort 기능을 사용할 수 있습니다.

# Get top 4 indices
n_max = 4
ind = np.argsort(a)[-n_max:]

# Retrieve top 4 values
top_max = a[ind]

인덱스 정렬:

기본적으로 argpartition은 정렬되지 않은 인덱스를 반환합니다. 정렬된 인덱스가 필요한 경우 다음을 사용하세요:

ind[np.argsort(a[ind])]

시간 복잡도:

• argpartition: 최악의 경우 O(n) 케이스
• argsort: O(n log n)
• 결합 접근법(정렬된 인덱스): 상위 k개 정렬 요소의 경우 O(n k log k)

위 내용은 NumPy 배열에서 여러 최대값의 인덱스를 찾는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!