Question

有没有办法在python中使用列表推导来从列表中过滤相邻的重复项？

这是我的意思的一个例子：

>>> xs = [1,2,2,3]
>>> print added.reAdj(xs)
[1,2,3]

通过SE搜索显示earlier inquiry询问类似但问题略有不同：是否可以从列表中删除所有重复项但未明确要求涉及列表的解决方案推导即可。使用列表推导的动机特别遵循对their advantages over traditional for loops的认可。用户建议使用set（）函数或标准循环：

result = []
most_recent_elem = None
for e in xs:
    if e != most_recent_elem:
        result.append(e)
        most_recent_elem = e

set()建议无法完成任务，因为删除了不相邻的重复项，而循环有效但冗长。

这似乎是一种安全地引用列表推导中的下一个元素的方法，如下所示。

[x for x in xs if x != **x.next()**]

有什么想法吗？

Answer 1

您可以使用itertools.groupby：

>>> import itertools
>>> [key for key, grp in itertools.groupby([1, 2, 2, 3])]
[1, 2, 3]

itertools.groupby返回一个迭代器。通过迭代，您将获得一个关键的组对。（key如果未指定key函数，则为项目，否则为key函数的返回值。 group是一个迭代器，它将生成通过应用key函数分组的项目（如果未指定，则将对相同的值进行分组）

>>> import itertools
>>> it = itertools.groupby([1, 2, 2, 3])
>>> it
<itertools.groupby object at 0x7feec0863048>
>>> for key, grp in it:
...     print(key)
...     print(grp)
... 
1
<itertools._grouper object at 0x7feec0828ac8>
2
<itertools._grouper object at 0x7feec0828b00>
3
<itertools._grouper object at 0x7feec0828ac8>
>>> it = itertools.groupby([1, 2, 2, 3])
>>> for key, grp in it:
...     print(list(grp))
... 
[1]
[2, 2]
[3]

以上解决方案，我只使用key因为问题并不关心相邻的项目数量。

Answer 2

您可以将list comprehension和enumerate与@AChampion建议的解决方案一起使用：

xs = [1,2,2,2,1,1]
In [115]: [n for i, n in enumerate(xs) if i==0 or n != xs[i-1]]
Out[115]: [1, 2, 1]

如果列表理解返回项是第一个，则返回项，如果它不等于之前，则返回以下项。它会因if语句的懒惰评估而起作用。

Answer 3

使用itertools配方中的pairwise（使用zip_longest）可以轻松检查下一个元素：

import itertools as it

def pairwise(iterable):
    a, b = it.tee(iterable)
    next(b, None)
    return it.zip_longest(a, b, fillvalue=object())   # izip_longest for Py2

>>> xs = [1,2,2,3]
>>> [x for x, y in pairwise(xs) if x != y]
[1, 2, 3]
>>> xs = [1,2,2,2,2,3,3,3,4,5,6,6]
>>> [x for x, y in pairwise(xs) if x != y]
[1, 2, 3, 4, 5, 6]

Answer 4

你可以使用一个不那么详细的循环解决方案：

>>> result = xs[:1]
>>> for e in xs:
        if e != result[-1]:
            result.append(e)

或者：

>>> result = []
>>> for e in xs:
        if e not in result[-1:]:
            result.append(e)

Answer 5

这个怎么样：

>>> l = [1,1,2,3,4,4,4,4,5,6,3,3,5,5,7,8,8,8,9,1,2,3,3,3,10,10]
>>> 
>>> o = []
>>> p = None
>>> for n in l:
        if n == p:
            continue
        o.append(n)
        p = n    

>>> o
[1, 2, 3, 4, 5, 6, 3, 5, 7, 8, 9, 1, 2, 3, 10]

显然，上面的解决方案比OP更详细，所以这里有一个替代使用来自itertools模块的zip_longest：

>>> l
[1, 1, 2, 3, 4, 4, 4, 4, 5, 6, 3, 3, 5, 5, 7, 8, 8, 8, 9, 1, 2, 3, 3, 3, 10, 10]
>>> from itertools import zip_longest
>>> o = [p for p,n in zip_longest(l,l[1:]) if p != n] #By default fillvalue=None
>>> o
[1, 2, 3, 4, 5, 6, 3, 5, 7, 8, 9, 1, 2, 3, 10]

如何使用列表推导删除列表中的相邻重复元素？

5 个答案: