提取第二层嵌套字典中的所有键

Question

我想提取 2d 字典第二级中的所有键，但 python 解释器返回 NameError。我的预期结果是 ['aa', 'bb', 'cc', 'aaa', 'bbb', 'ccc']

>>> adict
defaultdict(<class 'dict'>, {'b': {'aaa': 444, 'ccc': 666, 'bbb': 555}, 'a': {'aa': 111, 'cc': 333, 'bb': 222}})

>>> all = [ele for ele in adict[ww].keys() for ww in ['a', 'b']]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'ww' is not defined

Answer 1

你很接近了。您只需要重新排序循环：

all = [ele for ww in ['a', 'b'] for ele in adict[ww] ]

要理解原因，请考虑如何编写普通for循环：

all = []
for ww in ['a', 'b']:
    for ele in adict[ww]:
        all.append(ele)

请注意，循环的顺序保持不变。另外，我已经删除了.keys()，这是没有必要的，因为dict默认情况下 a 的迭代发生在键上。

你也可以像乔恩·克莱门茨一样很棒，这样做：

In [265]: set().union(*adict.values())
Out[265]: {'aa', 'aaa', 'bb', 'bbb', 'cc', 'ccc'}

*adict.values()返回内部字典的列表，其键被解包然后添加到集合中。一些提示：

1. 不保证顺序（即使在 python3.6 上）

2. 重复项被删除