Python:获取连续至少3个元素的索引

Question

(这不是家庭作业，我在自学(

我有一本字典(实际上是一本Counter(，里面有一些计数数字(称之为n(以及这个数字发生了多少次(称为c(。即

my_dict = { n0: c0, n1: c1, ... , nx: cx }

如果n是连续的，我想将其转换为c的子列表。即如果

my_dict = { 0: 1, 1: 1, 4: 1, 5: 2, 6: 1 }

我想要

result = [ [ 1 , 1 ] , [ 1 , 2 , 1 ] ]

我可以在索引上使用for循环，但这不会是蟒蛇式的。然而，在这一点上，我使用了多个itertools函数，但仍然没有得到我想要的。

我能得到一个建议吗？

Answer 1

for循环可能是最简单的解决方案；迭代字典的键，每当键不等于最后一个键加1时，在输出中启动一个新列表，否则附加到当前列表：

my_dict = { 0: 1, 1: 1, 4: 1, 5: 2, 6: 1 }
l = []
last = max(my_dict.keys())
for k, v in my_dict.items():
if k != last + 1:
l.append([v])
else:
l[-1].append(v)
last = k

print(l)

输出：

[[1, 1], [1, 2, 1]]

Answer 2

函数式编程有时会很可怕：

import functools
my_dict = { 0: 1, 1: 1, 4: 1, 5: 2, 6: 1 }
sorted_items = sorted(my_dict.items())
initial_data = [[sorted_items[0][1]]]
remaining_data = (
{'value': value, 'should_merge': previous_key + 1 == key}
for (previous_key, previous_value), (key, value)  in
zip(sorted_items[1:], sorted_items[2:]) 
)
my_result = functools.reduce(
lambda initial, data:
initial[-1:] + [initial[-1] + [data['value']]] if data['should_merge'] else
initial + [[data['value']]],
remaining_data,
initial_data
)