6

我的问题可能有点难以理解,但实际上就是这样。我有一个看起来像这样的嵌套字典:

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)
4

1 回答 1

3

list1设置并使用字典视图和列表理解:

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

在 Python 3 中,使用value.keys()anddict_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作品),但将其设置为更清楚地说明您的意图。

于 2013-05-13T08:53:48.343 回答