finden Sie Abschnitte von Trues in einem Array

Question

Gibt es eine gute Möglichkeit, Abschnitte von Trues in einem booleschen Array zu finden? Wenn ich ein Array haben wie:

x = numpy.array([True,True,False,True,True,False,False]) 

Kann ich eine Reihe von Indizes wie:

starts = [0,3] 
ends = [1,4] 

oder andere geeignete Weg, um diese Informationen zu speichern. Ich weiß, dass dies mit einigen komplizierten While-Loops möglich ist, aber ich suche nach einem besseren Weg.

Answer 1

Sie können x mit Falses (eins am Anfang und eins am Ende) padeln, und np.diff verwenden. Ein "diff" von 1 bedeutet einen Übergang von False nach True und von -1 bedeutet einen Übergang von True zu False.

Die Konvention soll das Ende des Bereichs als Index eins nach das letzte darstellen. Dieses Beispiel entspricht der Konvention (Sie leicht ends-1 statt ends verwenden können, um das Array in Frage zu kommen):

x1 = np.hstack([ [False], x, [False] ]) # padding 
d = np.diff(x1.astype(int)) 
starts = np.where(d == 1)[0] 
ends = np.where(d == -1)[0] 
starts, ends 
=> (array([0, 3]), array([2, 5]))