我对以下问题感到好奇:Eliminate consecutive duplicates of list elements,以及如何在Python中实现它。
我想到的是:
list = [1,1,1,1,1,1,2,3,4,4,5,1,2]
i = 0
while i < len(list)-1:
if list[i] == list[i+1]:
del list[i]
else:
i = i+1
输出:
[1, 2, 3, 4, 5, 1, 2]
我猜没关系。
所以我很好奇,想看看是否可以删除具有连续重复项的元素并获得以下输出:
[2, 3, 5, 1, 2]
为此,我这样做:
list = [1,1,1,1,1,1,2,3,4,4,5,1,2]
i = 0
dupe = False
while i < len(list)-1:
if list[i] == list[i+1]:
del list[i]
dupe = True
elif dupe:
del list[i]
dupe = False
else:
i += 1
但是似乎有点笨拙而不是pythonic,您是否有任何更聪明/更优雅/更有效的方法来实现这一点?
最佳答案
>>> L = [1,1,1,1,1,1,2,3,4,4,5,1,2]
>>> from itertools import groupby
>>> [x[0] for x in groupby(L)]
[1, 2, 3, 4, 5, 1, 2]
如果愿意,可以使用map代替列表理解
>>> from operator import itemgetter
>>> map(itemgetter(0), groupby(L))
[1, 2, 3, 4, 5, 1, 2]
对于第二部分
>>> [x for x, y in groupby(L) if len(list(y)) < 2]
[2, 3, 5, 1, 2]
如果不想创建临时列表只是为了获取长度,则可以在生成器表达式上使用sum
>>> [x for x, y in groupby(L) if sum(1 for i in y) < 2]
[2, 3, 5, 1, 2]