Блог

Главная - Вопросы по программированию - Объединить два одинаковых значения с одинаковыми ключами в списке словарей

Объединить два одинаковых значения с одинаковыми ключами в списке словарей

#python #list #dictionary

#python #Список #словарь

Вопрос:

Полностью новичок в программировании .. и нужна помощь.

У меня есть список словарей в следующем формате:

 a_list = [
    {'ID': 'a', 'Animal': 'dog', 'color': 'white', 'tail': 'yes'},
    {'ID': 'a', 'Animal': 'cat', 'color': 'black', 'tail': 'yes'},
    {'ID': 'b', 'Animal': 'bird', 'color': 'black', 'tail': 'no'},
    {'ID': 'b', 'Animal': 'cat', 'color': 'pink', 'tail': 'yes'}
    {'ID': 'b', 'Animal': 'dog', 'color': 'yellow', 'tail': 'no'}      
   ]

То, что я ищу, — это список словарей следующим образом:

 a_dict = 
    {'a': {'dog': {'color': 'white', 'tail': 'yes'},
           'cat': {'color': 'black', 'tail': 'yes'}},
     'b': {'bird': {'color': 'black', 'tail': 'no'},
           'cat': {'color': 'pink', 'tail': 'no'},
           'dog': {'color': 'yellow', 'tail': 'no'}}}

Комментарии:

1. Что вы пробовали? С чем вы боретесь?

2. Что вы пробовали до сих пор?

3. Я попытался for loop использовать идентификатор, чтобы получить список уникальных идентификаторов. Затем я pop извлекаю идентификатор из исходного списка. Я объединяю два списка с. zip Затем я обнаружил, zip что может быть только один уникальный ключ к одному значению… Не одинаковые ключи (= один и тот же уникальный ключ) с разными значениями. @ThomasSablik @SurajSubramanian

Ответ №1:

 a_list = [
    {'ID': 'a', 'Animal': 'dog', 'color': 'white', 'tail': 'yes'},
    {'ID': 'a', 'Animal': 'cat', 'color': 'black', 'tail': 'yes'},
    {'ID': 'b', 'Animal': 'bird', 'color': 'black', 'tail': 'no'},
    {'ID': 'b', 'Animal': 'cat', 'color': 'pink', 'tail': 'yes'},
    {'ID': 'b', 'Animal': 'dog', 'color': 'yellow', 'tail': 'no'}      
]

a_dict = {}
for v in a_list:
    a_dict.setdefault(v['ID'], {}).setdefault(v['Animal'], {}).update(color=v['color'], tail=v['tail'])

from pprint import pprint
pprint(a_dict)

С принтами:

 {'a': {'cat': {'color': 'black', 'tail': 'yes'},
       'dog': {'color': 'white', 'tail': 'yes'}},
 'b': {'bird': {'color': 'black', 'tail': 'no'},
       'cat': {'color': 'pink', 'tail': 'yes'},
       'dog': {'color': 'yellow', 'tail': 'no'}}}

Ответ №2:

Создал вложенный словарь и удалил из него все ключи, которые не понадобятся позже.

 a_list = [
    {'ID': 'a', 'Animal': 'dog', 'color': 'white', 'tail': 'yes'},
    {'ID': 'a', 'Animal': 'cat', 'color': 'black', 'tail': 'yes'},
    {'ID': 'b', 'Animal': 'bird', 'color': 'black', 'tail': 'no'},
    {'ID': 'b', 'Animal': 'cat', 'color': 'pink', 'tail': 'yes'},
    {'ID': 'b', 'Animal': 'dog', 'color': 'yellow', 'tail': 'no'}      
   ]

a_dict = {}
for a in a_list:
    if a['ID'] in a_dict:
        a_dict[a['ID']][a['Animal']] = a
    else:
        a_dict[a['ID']] = {a['Animal']: a}

for id_ in a_dict:
    for animal in a_dict[id_]:
        del a_dict[id_][animal]['ID']
        del a_dict[id_][animal]['Animal']

Вывод :

 >> a_dict

{'a': {'dog': {'color': 'white', 'tail': 'yes'},
  'cat': {'color': 'black', 'tail': 'yes'}},
 'b': {'bird': {'color': 'black', 'tail': 'no'},
  'cat': {'color': 'pink', 'tail': 'yes'},
  'dog': {'color': 'yellow', 'tail': 'no'}}}

Ответ №3:

Простое решение с defaultdict 

 from collections import defaultdict
result = defaultdict(dict)
a_list = [
    {'ID': 'a', 'Animal': 'dog', 'color': 'white', 'tail': 'yes'},
    {'ID': 'a', 'Animal': 'cat', 'color': 'black', 'tail': 'yes'},
    {'ID': 'b', 'Animal': 'bird', 'color': 'black', 'tail': 'no'},
    {'ID': 'b', 'Animal': 'cat', 'color': 'pink', 'tail': 'yes'},
    {'ID': 'b', 'Animal': 'dog', 'color': 'yellow', 'tail': 'no'}      
   ]
for item in a_list:
    result[item['ID']][item['Animal']] = {'color':item['color'], 'tail':item['tail']}
  
 defaultdict(dict,
            {'a': {'dog': {'color': 'white', 'tail': 'yes'},
              'cat': {'color': 'black', 'tail': 'yes'}},
             'b': {'bird': {'color': 'black', 'tail': 'no'},
              'cat': {'color': 'pink', 'tail': 'yes'},
              'dog': {'color': 'yellow', 'tail': 'no'}}})

Ответ №4:

Как и в других предлагаемых здесь решениях, в этом используется словарь по умолчанию.

 >>> from collections import defaultdict
>>> d = defaultdict(dict)

>>> for dict_ in a_list:
    ID = dict_.pop('ID')
    animal = dict_.pop('Animal')
    d[ID][animal] = dict_

>>> from pprint import pprint
>>> pprint(dict(d))
{'a': {'cat': {'color': 'black', 'tail': 'yes'},
       'dog': {'color': 'white', 'tail': 'yes'}},
 'b': {'bird': {'color': 'black', 'tail': 'no'},
       'cat': {'color': 'pink', 'tail': 'yes'},
       'dog': {'color': 'yellow', 'tail': 'no'}}}

Ответ №5:

Вы можете создавать словарь итеративно, создавая внутренние словари на каждом шаге. Вот реализация предложения для общего случая:

 def group_by_nested_keys(elements, keys):
    root = {}
    for element in elements:
        inner_dict = root
        # Create the inner dictionaries as we go through the keys
        # and remove the key from element at the same time 
        for key in keys[:-1]:
            inner_dict = inner_dict.setdefault(element.pop(key), {})

        # When all the dictionaries for the elements are created,
        # assign the element to the most inner dictionary
        inner_dict[element.pop(keys[-1])] = element

    return root



a_dict = group_by_nested_keys(a_list, ["ID", "Animal"])