比较字典忽略特定的键

如何测试两个字典是否相同,同时考虑一些密钥。 例如,

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_

(请注意,我们不需要深层复制,我们只需要避免修改d1d2 。)

链接地址: http://www.djcxy.com/p/22553.html

上一篇: Compare dictionaries ignoring specific keys

下一篇: How can I sort a dictionary by key?