我有一个整数列表,我希望能够识别连续的重复块:也就是说,我想生成一个保序的双组列表,其中每个双组包含(int_in_question,出现次数)。
例如,如果我有一个类似的列表:
[0, 0, 0, 3, 3, 2, 5, 2, 6, 6]
我想要的结果是:
[(0, 3), (3, 2), (2, 1), (5, 1), (2, 1), (6, 2)]
我有一个相当简单的方法,使用 for 循环、临时变量和计数器来完成此操作:
result_list = []
current = source_list[0]
count = 0
for value in source_list:
if value == current:
count += 1
else:
result_list.append((current, count))
current = value
count = 1
result_list.append((current, count))
但我真的很喜欢 python 的函数式编程习惯,并且我希望能够使用简单的生成器表达式来完成此操作。然而,我发现在使用生成器时很难保持子计数。我有一种感觉,两步过程可能会让我到达那里,但现在我被难住了。
有没有一种特别优雅/Pythonic 的方法来做到这一点,尤其是使用生成器?
>>> from itertools import groupby
>>> L = [0, 0, 0, 3, 3, 2, 5, 2, 6, 6]
>>> grouped_L = [(k, sum(1 for i in g)) for k,g in groupby(L)]
>>> # Or (k, len(list(g))), but that creates an intermediate list
>>> grouped_L
[(0, 3), (3, 2), (2, 1), (5, 1), (2, 1), (6, 2)]
包括电池,正如他们所说。
使用 JBernardo 的
sum
和生成器表达式的建议;看评论。