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

a = {'joe': [24,32,422], 'bob': [1,42,32,24], 'jack':[0,3,222]}
b = {'joe': [24], 'bob': [1,42,32]}

Я хотел бы получить разницу между этими двумя словарями, которая в этом случае будет выглядеть так:

{'joe': [32,422], 'bob': [24], 'jack':[0,3,222]}

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

Я попробовал: a.items() - b.items()

Но я получаю следующую ошибку: unsupported operand type(s) for -: 'dict_values' and 'dict_values'

Спасибо за вашу помощь

12
Ryan 15 Дек 2015 в 23:18

3 ответа

Лучший ответ

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

>>> a = {'joe': [24,32,422], 'bob': [1,42,32,24], 'jack':[0,3,222]}
>>> b = {'joe': [24], 'bob': [1,42,32]}
>>> {key: list(set(a[key])- set(b.get(key,[]))) for key in a}
{'joe': [32, 422], 'bob': [24], 'jack': [0, 3, 222]}

Обратите внимание на две вещи:

  • Я преобразовываю набор обратно в список, когда я устанавливаю его как значение
  • Я использую b.get вместо b[key] для обработки, если ключ не существует в b, но существует в a

РЕДАКТИРОВАТЬ - используя цикл for:

Я понял, что понимание может быть не столь очевидным, так что это эквивалентный фрагмент кода с использованием цикла for:

>>> c = {}
>>> for key in a:
    c[key] = list(set(a[key]) - set(b.get(key,[])))


>>> c
{'joe': [32, 422], 'bob': [24], 'jack': [0, 3, 222]}

РЕДАКТИРОВАТЬ - потерять второй сет:

Как упоминал Падрейк Каннингем в комментариях (как он часто делает, благословляет его душу), вы можете использовать set.difference, чтобы избежать явного приведения вашего второго списка к набору:

>>> c = {}
>>> for key in a:
    c[key] = list(set(a[key]).difference(b.get(key,[])))


>>> c
{'joe': [32, 422], 'bob': [24], 'jack': [0, 3, 222]}

Или с пониманием списка:

>>> {key: list(set(a[key]).difference(b.get(key,[]))) for key in a}
{'joe': [32, 422], 'bob': [24], 'jack': [0, 3, 222]}

Или если вы хотите рассматривать set.difference как метод класса вместо метода экземпляра:

>>> {key: list(set.difference(set(a[key]),b.get(key,[]))) for key in a}
{'joe': [32, 422], 'bob': [24], 'jack': [0, 3, 222]}

Хотя я нахожу это немного неуклюжим, и мне это не очень нравится.

12
R Nar 15 Дек 2015 в 21:04

Другой способ - использование встроенного метода filter:

>>> a = {'joe': [24,32,422], 'bob': [1,42,32,24], 'jack':[0,3,222]}
>>> b = {'joe': [24], 'bob': [1,42,32]}
>>> {key:filter(lambda s: s not in b.get(key,[]), a[key]) for key in a}
{'bob': [24], 'joe': [32, 422], 'jack': [0, 3, 222]}

Согласно Padraic Cunningham комментариям:

В Python 3 filter возвращает generator, поэтому вам нужно преобразовать его в список следующим образом:

{key:list(filter(lambda s: s not in b.get(key,[]), a[key])) for key in a}

3
Iron Fist 15 Дек 2015 в 21:06

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

diff = {}
for key in a:
    diff[key] = list(set(a[key]) - set(b.get(key, [])))
print diff
7
Martin Valgur 15 Дек 2015 в 20:45