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 循环，但这不是 pythonic。然而，此时我正在使用多个 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
)

Python：获取连续至少 3 的元素的索引？

Python: get indices of elements that are in a streak of at least 3?

python

itertools