Wie finde ich effizient Zeilenindizes mit mehreren Werten in NumPy-Arrays?

Zeilenindizes mit mehreren Werten in NumPy-Arrays finden

Gegeben ein Array X und eine Reihe von Zielzeilen mit gesuchten_Werten besteht die Aufgabe darin, die entsprechenden Zeilenindizes abzurufen. Dieses Problem kann mithilfe verschiedener NumPy-Funktionen effizient gelöst werden.

Ansatz 1: Broadcasting

Ein einfacher Ansatz mithilfe von Broadcasting:

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

Ansatz 2: Speichereffiziente Konvertierung

Um die Speichereffizienz zu erhöhen, konvertieren Sie jede Zeile in einen eindeutigen linearen Index und verwenden Sie ihn 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]

Ansatz 3: Speichereffiziente Suche

Eine weitere speichereffiziente Lösung mit 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)]

Hinweis: Bei diesem Ansatz wird davon ausgegangen, dass jede Zeile in „searched_values“ eine Übereinstimmung in X aufweist.

Das obige ist der detaillierte Inhalt vonWie finde ich effizient Zeilenindizes mit mehreren Werten in NumPy-Arrays?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!