Znajdź liczbę kolejnych elementów, które są takie same, zanim zmieni

Question

Tak na przykład, jeśli mam list

a = [1,1,1,2,2] 
b = [1,1,2,2,2] 
c = [2,1,1,1,1] 

chciałbym uzyskać najdłuższą passę pierwszy element na liście, więc na przykład a dałoby 3, b dałoby 2, a c dałoby 1. Wiem, że mógłbym utworzyć pętlę while i policzyć pas w ten sposób, ale zastanawiałem się, czy istnieje bardziej elegancki sposób na zrobienie tego?

Answer 1

mógłby zrobić coś takiego:

numStreak = ([a[0] == n for n in a]+[False]).index(False) 

(w tym także dba o to, czy wszystkie elementy są jak pierwszy element, indeks nie zwróci poprawną wartość)

UPDATE: bardziej wydajny (ale mniej elegancki?) wersja

from itertools import takewhile 
len([1 for _ in takewhile(lambda x:x==a[0], a)]) 

lub nieco lepsza (UPDATE 2) @ vaultah za sugestia:

sum(1 for _ in takewhile(lambda x:x==a[0], a)) 

Answer 2

Można użyć GroupBy i podsumować ile przedmioty są w pierwszej grupie dla każdego:

a = [1,1,1,2,2] 
b = [1,1,2,2,2] 
c = [2,1,1,1,1] 

from itertools import groupby 

for l in [a,b,c]: 
    print(sum(1 for _ in next(groupby(l), [[], []],)[1])) 

lub używając takewhile:

from itertools import takewhile 

for l in [a, b, c]: 
    print(sum((1 for _ in takewhile(lambda x: x == l[0], l)))) 

Jeśli dane jest zawsze lista, krotka etc. W grupie groupby można sprawdzić wartość falsey w przeciwieństwie do ustawienia domyślnego w next(..:

for l in [a, b, c]: 
    print(sum(1 for _ in next(groupby(l))[1]) if l else 0) 

Answer 3

Można użyć numpy:

>>> import numpy as np 
>>> a = [1,1,1,2,2] 
>>> b = [1,1,2,2,2] 
>>> c = [2,1,1,1,1] 
>>> def runs(data): 
... return np.split(data, np.where(np.diff(data) != 0)[0]+1) 
... 
>>> for e in a,b,c: 
... runs(np.array(e)) 
... 
[array([1, 1, 1]), array([2, 2])] 
[array([1, 1]), array([2, 2, 2])] 
[array([2]), array([1, 1, 1, 1])] 

Następnie wystarczy wziąć długość pierwszego biegu:

>>> for e in a,b,c: 
... len(runs(np.array(e))[0]) 
... 
3 
2 
1 

Lub, w Pythonie, wystarczy użyć while pętlę:

>>> def r(a): 
... i=1 
... while a[0]==a[i]: i+=1 
... return i 
... 
>>> r(a) 
3 
>>> r(b) 
2 
>>> r(c) 
1 

Answer 4

Jedna wkładka do drogi? Wycelować ...

a = [5,5,5,5,8] 
list(np.ediff1d(a)).count(0)+1 
>>> 4