Liste der Indizes von jedem Wert

Question

Gibt es eine einzige Numpy Funktion, die

[array == value for value in np.unique(array)] 

oder

[np.where(array == value) for value in np.unique(array)] 

Oder wenn nicht, wird ein effizienter Weg, dies zu tun, entspricht? Diese durchlaufen das Array len(np.unique(array)) Mal, wenn Sie es in einem einzigen Durchlauf tun können:

indices = defaultdict(list) 
for index, value in enumerate(array): 
    indices[value].append(index) 

Ich würde eine Lösung wie die keine expliziten erfordert für Schleife.

Answer 1

Wenn Sie was Sie wollen, ist eine Maske, mit mehreren eindeutigen Nummer Bedingungen, können Sie es return_inverse von np.unique tun mit:

eine Probe-Array mit

>>> a = np.random.randint(5, 10, size=100) # 100 [5-10) random numbers 

Und seine einzigartige Zahlen und inverse Abbildung

Wir können einen Container mit zusätzlichen Abmessungen (a.shape + 1 Dimension Feind jedes einzigartig erstellen Nummer)

>>> indexes = np.zeros((a.shape[0], unique.size), dtype=np.bool) 

und schließlich das Array mit der inversen Mapping-füllen:

>>> indexes[np.arange(a.size), inverse] = True 

Die indexes Karte True in der letzten Dimension enthält, um den Index der eindeutigen Nummer entspricht, die es übereinstimmt (die Reihenfolge im unique Array).

>>> indexes[:3, :] 
array([[False, False, False, False, True], 
     [ True, False, False, False, False], 
     [False, False, False, True, False]], dtype=bool) 

Jede Zeile entspricht dem Index des ursprünglichen Arrays a und jede Spalte entspricht einer eindeutigen Nummer .

Answer 2

Ihre Frage ist ein wenig mehrdeutig; aber sehen Sie sich numpy_indexed an; Es scheint, als ob entweder npi.indices, npi.contains oder npi.in_ zur Lösung Ihres Problems verwendet werden können.