Bagaimana untuk Mencari Indeks Baris Berbilang Nilai dengan Cekap dalam Tatasusunan NumPy?

Mencari Indeks Baris Berbilang Nilai dalam Tatasusunan NumPy

Memandangkan tatasusunan X dan satu set baris sasaran searched_values, tugasnya adalah untuk mendapatkan semula indeks baris yang sepadan. Masalah ini boleh diselesaikan dengan cekap menggunakan pelbagai fungsi NumPy.

Pendekatan 1: Penyiaran

Pendekatan mudah menggunakan penyiaran:

np.where((X==searched_values[:,None]).all(-1))[1]

Pendekatan 2: Penukaran Cekap Memori

Untuk kecekapan memori, tukar setiap baris kepada indeks linear yang unik dan gunakan np.in1d:

dims = X.max(0)+1
out = np.where(np.in1d(np.ravel_multi_index(X.T,dims),\
                    np.ravel_multi_index(searched_values.T,dims)))[0]

Pendekatan 3: Carian Cekap Memori

Satu lagi penyelesaian cekap memori menggunakan np.searchsorted:

dims = X.max(0)+1
X1D = np.ravel_multi_index(X.T,dims)
searched_valuesID = np.ravel_multi_index(searched_values.T,dims)
sidx = X1D.argsort()
out = sidx[np.searchsorted(X1D,searched_valuesID,sorter=sidx)]

Nota: Pendekatan ini menganggap bahawa setiap baris dalam searched_values ​​mempunyai padanan dalam X.

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