Переберите список словарей, найдите подходящие элементы из списка и добавьте значение соответствующего ключа в defaultdict - PullRequest
Новая
       11

Переберите список словарей, найдите подходящие элементы из списка и добавьте значение соответствующего ключа в defaultdict

0 голосов
/ 24 октября 2018

У меня есть список словарей.Давайте назовем это: list_of_dict.Словари в списке имеют вид: 

{'a' : 1,
'b' : 5,
'c' : 3,
'd' : 6}

и 

{'a' : 3,
'f' : 2,
'g' : 1,
'h' : 3,
'i' : 5,
'j' : 3}

У меня есть еще один список с именем list_to_match, который содержит только несколько букв: ['a', 'f', 'x']

Мне нужно перебрать list_of_dict и найти ключи, которые соответствуют элементу в списке, и добавить значения в пустой стандартный список элементов списка.Если файл не найден, добавьте к списку 0.

defaultdict инициализируется как: 

d = collections.defaultdict(list)

Я хочу, чтобы конечный вариант defaultdict выглядел следующим образом: 

{'a' : [1, 3],
'f' : [0, 2],
'x' : [0, 0]}

Пока что у меня есть: 

for ld in list_of_dict:
    for match in list_to_match:
        for k, v in ld.items():
            d[match].append(v)
    d[match].append(0)

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

{'a' : [1, 3],
'f' : [2]}

Отсутствует 0, равно как и x.Как мне это исправить?

Ответы [ 3 ]

0 голосов
/ 24 октября 2018

Вы можете использовать словарь понимания: 

{i: [j[i] if i in j else 0 for j in list_of_dict] for i in list_to_match}

Выход: 

{'a': [1, 3], 'f': [0, 2], 'x': [0, 0]}

Еще проще: 

{i: [j.get(i, 0) for j in list_of_dict] for i in list_to_match}
0 голосов
/ 24 октября 2018

IIUC, вы можете просто сделать: 

for m in list_to_match:
    d[m] = [ld.get(m, 0) for ld in list_of_dict]
print(d)
#defaultdict(list, {'a': [1, 3], 'f': [0, 2], 'x': [0, 0]})

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

0 голосов
/ 24 октября 2018

Вы можете сделать это, не нужно использовать defaultdict: 

list_of_dict = [{'a': 1, 'b': 5,
                 'c': 3,
                 'd': 6}, {'a': 3,
                           'f': 2,
                           'g': 1,
                           'h': 3,
                           'i': 5,
                           'j': 3}]

list_to_match = ['a', 'f', 'x']

d = {}
for match in list_to_match:
    for ld in list_of_dict:
        d.setdefault(match, []).append(ld.get(match, 0))

print(d)

Выход 

{'a': [1, 3], 'x': [0, 0], 'f': [0, 2]}
...