Theo dõi thay đổi giá trị trong một danh sách lặp đi lặp lại bằng Python

Question

tôi có một danh sách với các giá trị lặp đi lặp lại như hình dưới đây:

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

Danh sách này được tạo ra từ một mô hình phù hợp với biểu thức chính quy (không hiển thị ở đây). Danh sách được đảm bảo có giá trị lặp lại (nhiều, nhiều lần lặp lại - hàng trăm, nếu không phải hàng nghìn) và không bao giờ được sắp xếp ngẫu nhiên vì đó là những gì regex phù hợp với từng thời điểm.

Điều tôi muốn là theo dõi chỉ mục danh sách mà tại đó các mục nhập thay đổi từ giá trị trước đó. Vì vậy, đối với danh sách trên x, tôi muốn có danh sách theo dõi thay đổi [3, 6] cho biết rằng x[3] và x[6] khác với các mục nhập trước đó của chúng trong danh sách.

Tôi đã làm được điều này, nhưng tôi đã tự hỏi liệu có cách nào sạch hơn không. Dưới đây là mã của tôi:

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 

Output: [3, 6]

Câu hỏi: Có cách nào sạch hơn để làm những gì tôi muốn, trong ít dòng mã?

Answer 1

Nó có thể được thực hiện bằng cách sử dụng danh sách hiểu, với một chức năng 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

thay vì đa indexing có O(n) phức tạp, bạn có thể sử dụng một iterator để kiểm tra các phần tử tiếp theo trong danh sách:

>>> 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

Tôi ở đây để thêm câu trả lời bắt buộc có chứa một danh sách hiểu.

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

Answer 4

Bạn có thể làm một cái gì đó như thế này sử dụng itertools.izip, itertools.tee và một danh sách-hiểu:

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] 

Một thay thế sử dụng itertools.groupby trên enumerate(x). groupby nhóm mặt hàng tương tự với nhau, vì vậy tất cả chúng ta cần là chỉ số của mặt hàng đầu tiên của mỗi nhóm ngoại trừ thứ nhất:

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] 

Nếu NumPy là một lựa chọn:

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

Answer 5

itertools.izip_longest là những gì bạn đang tìm kiếm cho:

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)