比较字典忽略特定的键
如何测试两个字典是否相同,同时考虑一些密钥。 例如,
equal_dicts(
{'foo':1, 'bar':2, 'x':55, 'y': 77 },
{'foo':1, 'bar':2, 'x':66, 'z': 88 },
ignore_keys=('x', 'y', 'z')
)
应该返回True。
UPD:我正在寻找一种高效,快速的解决方案。
UPD2。 我结束了这个代码,这看起来是最快的:
def equal_dicts_1(a, b, ignore_keys):
ka = set(a).difference(ignore_keys)
kb = set(b).difference(ignore_keys)
return ka == kb and all(a[k] == b[k] for k in ka)
计时:https://gist.github.com/2651872
def equal_dicts(d1, d2, ignore_keys):
d1_filtered = dict((k, v) for k,v in d1.iteritems() if k not in ignore_keys)
d2_filtered = dict((k, v) for k,v in d2.iteritems() if k not in ignore_keys)
return d1_filtered == d2_filtered
编辑:这可能会更快,更具有内存效率:
def equal_dicts(d1, d2, ignore_keys):
ignored = set(ignore_keys)
for k1, v1 in d1.iteritems():
if k1 not in ignored and (k1 not in d2 or d2[k1] != v1):
return False
for k2, v2 in d2.iteritems():
if k2 not in ignored and k2 not in d1:
return False
return True
使用词典理解:
>>> {k: v for k,v in d1.items() if k not in ignore_keys} ==
... {k: v for k,v in d2.items() if k not in ignore_keys}
在Python 2上使用.viewitems()
。
非常非常粗鲁的是,你可以删除任何被忽略的键并比较这些字典:
def equal_dicts(d1, d2, ignore_keys=()):
d1_, d2_ = d1.copy(), d2.copy()
for k in ignore_keys:
try:
del d1_[k]
except KeyError:
pass
try:
del d2_[k]
except KeyError:
pass
return d1_ == d2_
(请注意,我们不需要深层复制,我们只需要避免修改d1
和d2
。)