去除具有连续重复的元素
我对这个问题很感兴趣:消除列表元素中的连续重复项,想知道在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
不过我觉得这个方法有点笨,不够优雅,你有没有更聪明、更优雅或更高效的实现方式?
9 个回答
15
如果你使用的是 Python 3.8 及以上版本,你可以使用一种叫做赋值表达式的东西,符号是 :=
:
list1 = [1, 2, 3, 3, 4, 3, 5, 5]
prev = object()
list1 = [prev:=v for v in list1 if prev!=v]
print(list1)
输出结果是:
[1, 2, 3, 4, 3, 5]
32
纯Python的一行代码
[v for i, v in enumerate(your_list) if i == 0 or v != your_list[i-1]]
103
>>> L = [1,1,1,1,1,1,2,3,4,4,5,1,2]
>>> from itertools import groupby
>>> [key for key, _group in groupby(L)]
[1, 2, 3, 4, 5, 1, 2]
对于第二部分
>>> [k for k, g in groupby(L) if len(list(g)) < 2]
[2, 3, 5, 1, 2]
如果你不想为了获取长度而创建一个临时列表,可以使用生成器表达式来进行求和。
>>> [k for k, g in groupby(L) if sum(1 for i in g) < 2]
[2, 3, 5, 1, 2]