У меня есть два списка, которые я хотел бы связать по индексу как пары ключевых значений в словаре. Список ключей содержит несколько идентичных элементов. Я хотел бы, чтобы все элементы в списке значений были спарены как список списка. Я использую метод list.append(), но это не дает мне желаемого результата. Любые рекомендации по коду, или я должен смотреть на проблему по-другому?
list1 = ['a', 'b', 'b', 'b', 'c']
list2 = [['1', '2', '3'], ['4', '5', '6'], [ '7', '8', '9'], ['10', '11', '12'], ['13', '14', '15']]
combo = {}
for i in range(len(list1)):
if list1[i] in combo:
combo[list1[i]].append(list2[i])
else:
combo[list1[i]] = list2[i]
Токовый выход:
{'a': ['1', '2', '3'], 'b': ['4', '5', '6', [ '7', '8', '9'], ['10', '11', '12']], 'c': ['13', '14', 15']}
Желаемый результат:
{'a': [['1', '2', '3']], 'b': [['4', '5', '6'], [ '7', '8', '9'], ['10', '11', '12']], 'c': [['13', '14', 15']]}
Использовать defaultdict
, с пустым списком начальное значение
result = defaultdict(list)
for key, value in zip(list1, list2):
result[key].append(value)
Если вы хотите получить более пифоновский ответ, вы также можете использовать расширение dict
:
output = {key: [value] for key, value in zip(list1, list2)}
Попробуйте этот код. Он работает, когда я пытаюсь использовать тот же ввод, который вы дали.
#Input
list1= ['a','b', 'b','b', 'c']
list2 = [['1', '2', '3'], ['4', '5', '6'], [ '7', '8', '9'], ['10','11','12'], ['13', '14', '15']]
combo= {}
for index, value in enumerate(list1):
if value in combo.keys():
combo[value].append(list2[i])
else:
combo[value]= []
combo.append(list2[i])
#output
print(combo)
{'a': [['1', '2', '3']],'b': [['4', '5', '6'], ['7', '8', '9'], ['10', '11', '12']], 'c': [['13', '14', '15']]}