valore pista cambiamenti in un elenco ripetitivo in Python

Question

Ho un elenco con valori ripetuti come mostrato di seguito:

x = [1, 1, 1, 2, 2, 2, 1, 1, 1] 

Questo elenco viene generato da un modello corrispondente espressione regolare (non illustrato qui). L'elenco è garantito per avere valori ripetuti (molte, molte ripetizioni - centinaia, se non migliaia), e non è mai organizzato in modo casuale perché è quello che la regex corrisponde ogni volta.

Quello che voglio è quello di monitorare gli indici della lista in cui le voci cambiano dal valore precedente. Pertanto, per l'elenco sopra riportato x, desidero ottenere un elenco di rilevamento delle modifiche [3, 6] che indica che x[3] e x[6] sono diversi dalle loro voci precedenti nell'elenco.

Sono riuscito a farlo, ma mi chiedevo se c'era un modo più pulito. Ecco il mio codice:

x = [1, 1, 1, 2, 2, 2, 1, 1, 1] 

flag = [] 
for index, item in enumerate(x): 
    if index != 0: 
     if x[index] != x[index-1]: 
      flag.append(index) 

print flag 

uscita: [3, 6]

Domanda: C'è un modo più pulito per fare quello che voglio, in un minor numero di righe di codice?

Answer 1

Esso può essere fatto utilizzando una lista di comprensione, con una funzione di range

>>> x = [1, 1, 1, 2, 2, 2, 3, 3, 3] 
>>> [i for i in range(1,len(x)) if x[i]!=x[i-1] ] 
[3, 6] 
>>> x = [1, 1, 1, 2, 2, 2, 1, 1, 1] 
>>> [i for i in range(1,len(x)) if x[i]!=x[i-1] ] 
[3, 6] 

Answer 2

invece multi-indicizzazione che ha O(n) complessità è possibile utilizzare un iteratore per verificare l'elemento successivo nella lista:

>>> x =[1, 1, 1, 2, 2, 2, 3, 3, 3] 
>>> i_x=iter(x[1:]) 
>>> [i for i,j in enumerate(x[:-1],1) if j!=next(i_x)] 
[3, 6] 

Answer 3

Sono qui per aggiungere la risposta obbligatoria che contiene una lista di comprensione.

flag = [i+1 for i, value in enumerate(x[1:]) if (x[i] != value)] 

Answer 4

si può fare qualcosa di simile utilizzando itertools.izip, itertools.tee e un elenco-di comprensione:

from itertools import izip, tee 
it1, it2 = tee(x) 
next(it2) 
print [i for i, (a, b) in enumerate(izip(it1, it2), 1) if a != b] 
# [3, 6] 

Un'altra alternativa utilizzando itertools.groupby su enumerate(x). groupby gruppi articoli simili insieme, quindi tutti abbiamo bisogno è l'indice della prima voce di ogni gruppo ad eccezione della prima:

from itertools import groupby 
from operator import itemgetter 
it = (next(g)[0] for k, g in groupby(enumerate(x), itemgetter(1))) 
next(it) # drop the first group 
print list(it) 
# [3, 6] 

Se NumPy è un'opzione:

>>> import numpy as np 
>>> np.where(np.diff(x) != 0)[0] + 1 
array([3, 6]) 

Answer 5

itertools.izip_longest è quello che stai cercando per:

from itertools import islice, izip_longest 

flag = [] 
leader, trailer = islice(iter(x), 1), iter(x) 
for i, (current, previous) in enumerate(izip_longest(leader, trailer)): 
    # Skip comparing the last entry to nothing 
    # If None is a valid value use a different sentinel for izip_longest 
    if leader is None: 
     continue 
    if current != previous: 
     flag.append(i)