检索嵌套字典的所有键(任意长度和深度)，其中键的名称决定了字典中的位置

Question

假设我有如下的字典:

{
'a1': {'b1': {'c1': 'val'}, 'b2': { 'c2': { 'd2': 'terminal'} } },
'a2': {'b2': 'val'}
}

我的目标是检索列表中的所有键:

'a1', 'b1', 'c1', 'b2', 'c2', 'd2', 'a2', 'b2'

，然后保留它们在字典中的位置，像这样:

'a1', 'a1.b1', 'a1.b1.c1', 'a1.b2', 'a1.b2.c2', 'a1.b2.c2.d2', 'a2', 'a2.b2'

这在Python

中是否可行?

Answer 1

一种方法是使用递归生成器函数:

data = {
'a1': {'b1': {'c1': 'val'}, 'b2': {'c2': {'d2': 'terminal'}}},
'a2': {'b2': 'val'}
}

def nested_iter(d, root=""):
for key, value in d.items():
printable = f"{root}.{key}" if root else key
if isinstance(value, dict):
yield printable
yield from nested_iter(value, root=printable)
else:
yield printable

print(list(nested_iter(data)))

['a1', 'a1.b1', 'a1.b1.c1', 'a1.b2', 'a1.b2.c2', 'a1.b2.c2.d2', 'a2', 'a2.b2']

作为一种选择，你可以使用collections.deque来处理递归:

def nested_iter_with_deque(d):
from collections import deque
result = []
q = deque(d.items())
while q:
k, vs = q.popleft()
result.append(k)
if isinstance(vs, dict):
lst = [(f"{k}.{key}", value) for key, value in vs.items()]
if lst:
q.extendleft(deque(lst))
return result

print(nested_iter_with_deque(data))

['a1', 'a1.b2', 'a1.b2.c2', 'a1.b2.c2.d2', 'a1.b1', 'a1.b1.c1', 'a2', 'a2.b2']