从列表中删除字典的子集
时间: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]