检查列表中是否存在列表中的项目

Question

我的问题可能有点复杂，但实际上这就是事情。 我有一个嵌套字典，如下所示：

dict_a = {'one': {'bird':2, 'tree':6, 'sky':1, 'TOTAL':9},
          'two': {'apple':3, 'sky':1, 'TOTAL':4},
          'three': {'tree':6, 'TOTAL':6},
          'four': {'nada':1, 'TOTAL':1},
          'five': {'orange':2, 'bird':3, 'TOTAL':5}
          }

和一份清单：

list1 = ['bird','tree']
newlist = []

如何检查list1中的项目是否在dict_a的嵌套字典中并将其附加到新列表？ 输出应如下所示：

newlist = ['one','three','five']

因为鸟和树恰好在一，三和五的嵌套字典中。

我能想到的是：

for s,v in dict_a.items():
    for s1,v1 in v.items():
        for item in list1:
            if item == s1:
               newlist.append(s)

Answer 1

使list1成为一个集合并使用字典视图和列表理解：

set1 = set(list1)
newlist = [key for key, value in dict_a.iteritems() if value.viewkeys() & set1]

在Python 3中，使用value.keys()和dict_a.items代替。

这将测试字典键和您要查找的键集之间是否存在集合交集（高效操作）。

演示：

>>> dict_a = {'one': {'bird':2, 'tree':6, 'sky':1, 'TOTAL':9},
...           'two': {'apple':3, 'sky':1, 'TOTAL':4},
...           'three': {'tree':6, 'TOTAL':6},
...           'four': {'nada':1, 'TOTAL':1},
...           'five': {'orange':2, 'bird':3, 'TOTAL':5}
...           }
>>> set1 = {'bird','tree'}
>>> [key for key, value in dict_a.iteritems() if value.viewkeys() & set1]
['three', 'five', 'one']

请注意，字典排序是任意的（取决于使用的键和字典插入和删除历史记录），因此输出列表顺序可能不同。

从技术上讲，你也可以直接使用你的列表（ value.viewkeys() & list1有效），但是把它作为一个集合可以更清楚地说明你的意图。