Создание словаря из подмножества списка
Предположим, у меня есть такой список Python:
master_list = ['a_1', 'b_4', 'd_2', 'c_3',
'a_2', 'c_1', 'd_4', 'b_3',
'd_3', 'c_2', 'a_4', 'b_1',
'c_4', 'a_3', 'b_1', 'd_1']
Каждый элемент в списке представляет собой букву, подчеркивание и цифру. Я хочу, чтобы словарь группировал эти элементы по номеру в следующем списке:
my_dict = {'1': [], '2': [], '3': [], '4':[]}
Хотя я мог бы использовать следующее для заполнения своих списков внутри my_dict, это кажется неудобным. Есть ли более эффективный способ добиться этого?
for item in master_list:
if '_1' in item:
my_dict['1'].append(item)
elif '_2' in item:
my_dict['2'].append(item)
elif '_3' in item:
my_dict['3'].append(item)
elif '_4' in item:
my_dict['4'].append(item)
@Woodford - я исправил свой вопрос и заголовок на основе ваших комментариев. Буду признателен, если вы вернете мне мою точку зрения. Здесь есть несколько хороших ответов.
Ответы 3
Отделите индекс и используйте его для добавления в правый список:
sortd_dict[item.split('_')[1]].append(item)
Вы можете использовать встроенную функцию split, чтобы разделить входные элементы по символу _.
sortd_dict = {}
def add_in_dict(k, v):
if k not in d:
sortd_dict[k] = [v]
else:
sortd_dict[k].append(v)
for item in master_list:
add_in_dict(item.split("_")[-1], item)
Все эти ответы сработали, но это решение и функция, которая разбивала его на k, v, казалось, работали лучше всего, когда я интегрировал их в свой код. Спасибо всем, что заглянули.
с groupby
from itertools import groupby
sort_func = lambda x: x[2]
master_list = sorted(master_list, key=sort_func)
res = {key: list(group) for (key, group) in groupby(master_list, key=sort_func)}
print(res)
Понимание и использование правильной терминологии поможет вам эффективно общаться с другими людьми и значительно улучшить качество результатов поиска. Здесь вы не создаете вложенный список — просто словарь. И мы не говорим о словарях как о сортировке или нет, поскольку они представляют собой неупорядоченный тип данных (за необычным исключением порядка вставки, имеющего значение в очень небольшом проценте случаев).