У меня есть список целых чисел, и я хочу иметь возможность идентифицировать непрерывные блоки дубликатов: т.е. я хочу создать список дубликатов, в котором каждый дубликат содержит (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, loop и counter:
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)]
Батареи включены, как говорится.
Предложение использования выражения sum и генератора от JBernardo; см. комментарий.