Rumah > pembangunan bahagian belakang > Tutorial Python > Bagaimana untuk Mencari Indeks Berbilang Nilai Maksimum dalam Tatasusunan NumPy?

Bagaimana untuk Mencari Indeks Berbilang Nilai Maksimum dalam Tatasusunan NumPy?

Linda Hamilton
Lepaskan: 2024-12-29 18:08:14
asal
306 orang telah melayarinya

How to Find the Indices of Multiple Maximum Values in a NumPy Array?

Mengambil semula Indeks Berbilang Nilai Maksimum dalam Tatasusunan NumPy

Tatasusunan NumPy menyediakan fungsi np.argmax untuk mencari indeks unsur maksimum . Walau bagaimanapun, jika anda memerlukan indeks nilai maksimum N, pertimbangkan penyelesaian berikut:

Versi NumPy Terkini:

Untuk NumPy versi 1.8 dan ke atas, fungsi argpartition menawarkan kaedah yang cekap:

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]
Salin selepas log masuk

NumPy Lama Versi:

Sebelum NumPy 1.8, anda boleh menggunakan fungsi argsort seperti berikut:

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

# Retrieve top 4 values
top_max = a[ind]
Salin selepas log masuk

Isih Indeks:

Secara lalai, argpartition mengembalikan indeks yang tidak diisih. Jika anda memerlukan indeks yang diisih, gunakan:

ind[np.argsort(a[ind])]
Salin selepas log masuk

Kerumitan Masa:

  • argpartition: O(n) dalam keadaan paling teruk kes
  • argsort: O(n log n)
  • Pendekatan gabungan (indeks diisih): O(n k log k) untuk elemen diisih atas-k

Atas ialah kandungan terperinci Bagaimana untuk Mencari Indeks Berbilang Nilai Maksimum dalam Tatasusunan NumPy?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!

sumber:php.cn
Kenyataan Laman Web ini
Kandungan artikel ini disumbangkan secara sukarela oleh netizen, dan hak cipta adalah milik pengarang asal. Laman web ini tidak memikul tanggungjawab undang-undang yang sepadan. Jika anda menemui sebarang kandungan yang disyaki plagiarisme atau pelanggaran, sila hubungi admin@php.cn
Artikel terbaru oleh pengarang
Tutorial Popular
Lagi>
Muat turun terkini
Lagi>
kesan web
Kod sumber laman web
Bahan laman web
Templat hujung hadapan