Frage:

Finde die Anzahl der ersten aufeinanderfolgenden True s
Betrachte a

Antwort 1
numpy : Benutze np.logical_and.accumulate für die Negation von a und nimm die Negation, um eine Maske zu erstellen, die die erste Reihe von False s eliminiert wenn sie existieren sollten. Dann fügen Sie am Ende ein False hinzu, um sicherzustellen, dass wir ein nicht True min haben. Verwenden Sie schließlich np.argmin , um den ersten Mindestwert zu finden. Wenn es eine Position 3 gefunden hat, zeigt das 3 True Werte davor an.

Antwort 2
numba.njit

Ich möchte numba verwenden, damit ich eine Schleife machen kann und sicherstellen kann, dass ich einen Kurzschluss erhalte, wenn wir wollen / müssen. Dies ist ein Problem, das sicher früh im Array beantwortet wird. Es ist nicht notwendig, die Dinge entlang des gesamten Arrays ohne Grund zu bewerten.

Antwort 3
numpy intelligentere Verwendung von argmin und argmax . Ich umschließe a mit False und dann argmax , um die erste True zu finden. Von diesem Punkt an, benutze argmin , um danach die erste False zu finden.
Hinweis: @Divakar hat eine Verbesserung dieser Antwort vorgenommen, die die Verwendung von np.concatenate eliminiert und stattdessen if/then/else verwendet. Das hat diese schon sehr schnelle Lösung um einen Faktor von 3 gekappt!

Wie schnell sind diese Antworten?
Siehe @ Divakars Antwort für den Quellcode anderer zeitgesteuerter Funktionen

Mit NumPy-Funktionen wäre eine Lösung -

Beispiellauf -

Für die Leistung müssen wir np.concatenate anstelle von np.r_ verwenden und dann die letzte Differenzierung durchschneiden -

Inspiriert von @ piRSquared's argmax und argmin trickery , hier ist noch eins mit einem Haufen IF-ELSE 's -

Versuchen Sie es auf diese Weise, die ersten aufeinanderfolgenden Vorkommen für True oder False und nur für True :

Ausgabe:

Ausgabe: