Cod*_*key 2 python dictionary list
我试图压平一个杂乱无章的字典(这又是从一个 json 文件中提取的)以简化提取信息。下面是字典的结构示例以及我尝试将其展平的示例:
data = {'horse':{'speed':{"walk": 40, "run":50}}, 'dog':{'run':30}, 'human':{'gait':{'normal':{'run': 25, 'walk': 30}}}}
flat_dict = []
for items in list(data.items()):
flat_list = []
flat_list.append(items[0])
try:
for item in list(items[1].items())[0]:
if type(item) is not dict:
flat_list.append(item)
else:
flat_list.append(list(item.keys())[0])
flat_list.append(list(item.values())[0])
except:
flat_list.append(items[0])
flat_dict.append(flat_list)
print(flat_dict)
Run Code Online (Sandbox Code Playgroud)
然而上面的代码并没有使整个字典变平并且丢失了一些信息,下面是上面代码的输出:
[['horse', 'speed', 'walk', 40], ['dog', 'run', 30], ['human', 'gait', 'normal', {'run': 25, 'walk': 30}]]
Run Code Online (Sandbox Code Playgroud)
我想要的是:
[['horse', 'speed', 'walk', 40, 'run', 50], ['dog', 'run', 30], ['human', 'gait', 'normal', 'run', 25, 'walk', 30]]
Run Code Online (Sandbox Code Playgroud)
我该怎么办?
您可以使用具有列表理解的递归方法:
def gen(d):
if isinstance(d, dict):
for k, v in d.items():
yield k
yield from gen(v)
else:
yield d
[[k, *gen(v)] for k, v in data.items()]
Run Code Online (Sandbox Code Playgroud)
输出:
[['horse', 'speed', 'walk', 40, 'run', 50],
['dog', 'run', 30],
['human', 'gait', 'normal', 'run', 25, 'walk', 30]]
Run Code Online (Sandbox Code Playgroud)