Python: получить индексы элементов, которые находятся в полосе не менее 3?
(это не домашнее задание, я сам учу)
У меня есть словарь (на самом деле Counter) с некоторыми счетными числами (назовем это n) и сколько раз это число встречалось (назовем это c). то есть
my_dict = { n0: c0, n1: c1, ... , nx: cx }
Я хочу преобразовать это в подсписки c, если n являются последовательными. то есть если
my_dict = { 0: 1, 1: 1, 4: 1, 5: 2, 6: 1 }
Я хотел бы
result = [ [ 1 , 1 ] , [ 1 , 2 , 1 ] ]
Я мог бы просто использовать цикл for для индекса, но это не было бы pythonic. Однако на данный момент я использую несколько функций itertools и до сих пор не получил то, что хочу.
Могу ли я получить предложение?
единственный способ, который я придумал, был очень непитоновским for i in range...
Я только что заметил: каково значение «серии не менее 3» в вашем названии? Ваш фактический вопрос демонстрирует захват всех длин полос (предполагая, что под «полосой» вы подразумеваете, что индексы последовательны).
Ответы 2
Цикл 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]]
Функциональное программирование иногда может быть ужасным:
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
)
forциклы могут быть питоническими...