从列表中删除字典的子集

时间:2020-03-06 15:04:40  来源:igfitidea点击:

这实际上仅是一个示例,很容易解释,因此要从字典中删除列表的交集,我通常会执行以下操作:

a = {1:'', 2:'', 3:'', 4:''}
exclusion = [3, 4, 5]

# have to build up a new list or the iteration breaks
toRemove = []
for var in a.iterkeys():
    if var in exclusion:
        toRemove.append(var)

for var in toRemove:
    del a[var]

这似乎是一个不寻常的例子,但是令人惊讶的是我不得不做这样的事情的次数。用集合来做会更好,但是我显然想保留字典的"值"。

这种方法很烦人,因为它需要两个循环和一个额外的数组。有没有更清洁,更有效的方法来执行此操作。

解决方案

考虑dict.pop:

for key in exclusion:
     a.pop(key, None)

key不是键时,None不会弹出异常。

为什么不只使用keys方法,而不是iterkeys?这样,我们可以在一个循环中完成此操作,因为它返回一个列表,而不是迭代器。

a = dict((key,value) for (key,value) in a.iteritems() if key not in exclusion)

我们可以将排除列表更改为一组,然后使用交集获得重叠。

exclusion = set([3, 4, 5])

for key in exclusion.intersection(a):
    del a[key]