从列表中删除dict的子集

时间:2008-10-03 14:24:35

标签: python list containers

这只是一个例子很容易解释,所以要从字典中删除列表的交集,我通常会这样做:

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的“值”。

这种方法很烦人,因为它需要两个循环和一个额外的数组。是否有更清洁,更有效的方法。

4 个答案:

答案 0 :(得分:12)

考虑dict.pop

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

None不是密钥时,key不会引发异常。

答案 1 :(得分:3)

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

答案 2 :(得分:2)

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

答案 3 :(得分:1)

您可以将排除列表更改为集合,然后只使用交集来获取重叠。

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

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