У меня есть список, который выглядит так:
l1 = ['200:200', '90:728']
У меня есть словарь, который выглядит так:
d1 = {'200:200':{'foo':'bar'},'300:300':{'foo':'bar'}}
Мне нужно отфильтровать дикториал, где только ключи находятся в l1. Диктофон должен выглядеть так:
result = {'200:200':{'foo':'bar'}}
В сущности, пересечение списка и ключей dict при возврате подраздела dict.
Как мне сделать это эффективно, когда время является проблемой для больших наборов?
Спасибо
Вы можете использовать следующий код:
keys = set(l1).intersection(set(d1.keys()))
result = {k:d1[k] for k in keys}
EDIT:. Как говорят комментаторы, вы можете заменить первую строку на Python 2.x:
keys = set(l1).intersection(d1)
И в Python 3.x:
keys = d1.keys() & l1
В 3.x это может быть просто:
>>> {k: d1[k] for k in (d1.keys() & l1)}
{'200:200': {'foo': 'bar'}}
В разделе 2.7 вы можете использовать dict.viewkeys() для воссоздания этой функции:
>>> {k: d1[k] for k in (d1.viewkeys() & l1)}
{'200:200': {'foo': 'bar'}}
В более старых версиях 2.x это несколько более подробный:
>>> {k: d1[k] for k in (set(d1).intersection(l1))}
{'200:200': {'foo': 'bar'}}
Не уверен в производительности каждого решения, но я бы сделал:
{k: v for k, v in d1.items() if k in l1}
Вы можете использовать понимание списка в конструкторе dict:
result = dict([(k,d1[k]) for k in l1 if k in d1])
Если вы беспокоитесь об удалении повторяющихся ключей, сначала сделайте l1 в наборе:
result = dict([(k,d1[k]) for k in set(l1) if k in d1])
Определите эффективность. В любом случае, что бы я сделал. Если бы это было слишком медленно, я бы, вероятно, переместил его в Cython.
s1 = set(l1)
s2 = set(d1.keys())
s3 = s1 & s2
# now you can access d1 using only keys in s3, or construct a new dict if you like
d2 = dict([(k,d1[k]) for k in s3])
Если распределение памяти и освобождение делают этот процесс слишком длинным, itertools на помощь.
import itertools
result = {dict_key:d1[dict_key] for dict_key in itertools.ifilter(lambda list_item: list_item in d1, l1) }
Это лишний раз выделяет память для всей новой коллекции, а l1 легко может быть итератором вместо списка.