使用 yield 压平嵌套字典有多简单

第一个博客

使用 yield 压平嵌套字典有多简单

你好！这是我的第一个博客
使用yield可以简化对字典多层嵌套的操作，相比递归的话会代码会更加简短！

这是我们经常用到的一个字典

nest_dict = {
‘a’: 1,
‘b’: {
‘c’: 2,
‘d’: 3,
‘e’: {‘f’: 4}
},
‘g’: {‘h’: 5},
‘i’: 6,
‘j’: {‘k’: {‘l’: {‘m’: 8}}}
}

这是我们想要修得到的字典

{
‘a’: 1,
‘b_c’: 2,
‘b_d’: 3,
‘b_e_f’: 4,
‘g_h’: 5,
‘i’: 6,
‘j_k_l_m’: 8
}

代码

def flat(x):
for key, value in x.items():
if isinstance(value, dict):
for k, v in flat(value):
k = f’{key}_{k}’
yield (k, v)
else:
yield (key, value)
new_dict={k:v for k,v in flat(nest_dict)}
print(new_dict)

isinstance()函数
isinstance(参数，类型）返回 Ture或Flase

**yield() **

现在我们可以来谈谈yield了，其实yield就相当于一个return，只是return返回的是值，但是yield返回的是生成器，除了*这点其他都一样，所以return也好yield也好都只能用在函数中，不要出现这种代码了

结果：

{‘a’: 1, ‘b_c’: 2, ‘b_d’: 3, ‘b_e_f’: 4, ‘g_h’: 5, ‘i’: 6, ‘j_k_l_m’: 8}