将一个字典与另一个字典进行比较，并选择仅存在于后者中的键。

Question

我有两个字典：

union = {'a':'gamma','b':'beta','d':'theta'}
packaged = {'a':'alpha','b':'gamma','c':'alpha'}

I want:

packaged = {'a': 'alpha', 'b': 'gamma'}

因此，我希望从packaged中仅选择那些与其键匹配的union的键，以及它们的值。

我阅读了类似的问题，并进行了以下操作：

for k, v in list(packaged.items()):
    if k not in union.keys():
        del packaged[k]
print(packaged)

这给了我想要的答案。

我的方法是最快/高效的吗？如果不是，有更快/高效的方法吗？

- R4444

Answer 1

你可以使用字典推导式。这会创建一个新的字典，但其复杂度与你的for循环相同：

d = {k: v for k, v in packaged.items() if k in union}

下面是一些真实的基准测试，以及两个稍微更高效的变体：

union = {'a':'gamma','b':'beta','d':'theta', **dict.fromkeys(range(5000, 10000))}
packaged = {'a':'alpha', 'b':'gamma', 'c':'alpha', **dict.fromkeys(range(1, 15000))}

def dct_cmp(union, packaged):
    return {k: v for k, v in packaged.items() if k in union}

def dct_cmp_from_key(union, packaged):
    return {k: packaged[k] for k in packaged if k in union}

def dct_cmp_from_key_intersect(union, packaged):
    return {k: packaged[k] for k in set(packaged) & set(union)}

def loopy(union, packaged):
    for k, v in list(packaged.items()):
        if k not in union:
            del packaged[k]
    return packaged

assert dct_cmp(union, packaged.copy()) == loopy(union, packaged.copy())
assert dct_cmp(union, packaged.copy()) == dct_cmp_from_key(union, packaged.copy())
assert dct_cmp(union, packaged.copy()) == dct_cmp_from_key_intersect(union, packaged.copy())

%timeit dct_cmp(union, packaged.copy())                     # 1.94 ms
%timeit dct_cmp_from_key(union, packaged.copy())            # 1.8 ms
%timeit dct_cmp_from_key_intersect(union, packaged.copy())  # 1.8 ms
%timeit loopy(union, packaged.copy())                       # 2.75 ms

- jpp

---

也许值得测试 d = {k: packaged[k] for k in packaged if k in union}，因为它不会从 items() 创建副本。 - RoadRunner

@RoadRunner，好的，已经添加了一个基于“交叉口”的解决方案。 - jpp