从列标题中删除整数列表键

Question

我有一个Python脚本，它使用了以前的Stack Overflow解决方案中的一个函数。

from pandas import json_normalize
from collections.abc import MutableMapping as mm
def flatten(dictionary, parent_key=False, separator='.'):
items = []
for key, value in dictionary.items():
new_key = str(parent_key) + separator + key if parent_key else key
if isinstance(value, mm):
items.extend(flatten(value, new_key, separator).items())
elif isinstance(value, list):
for k, v in enumerate(value):
items.extend(flatten({str(k): v}, new_key).items())
else:
items.append((new_key, value))
return dict(items)
d = { 
"_id" : 1, 
"labelId" : [
6422
], 
"levels" : [
{
"active" : "true", 
"level" : 3, 
"actions" : [
{
"isActive" : "true"
}]
}]
}
x = flatten(d)
x = json_normalize(x)
print(x)

电流输出：

_id  labelId.0 levels.0.active  levels.0.level levels.0.actions.0.isActive 
0    1       6422            true               3                        true                           

我遇到的问题是列名中包含的数字键。有没有一种方法可以修改我的代码以实现我想要的输出？

所需输出：

_id    labelId   levels.active    levels.level     levels.actions.isActive 
0    1       6422            true               3                        true                           

Answer 1

首先，将parent_key用作布尔，然后为其分配其他类型值不是最佳做法。它有效，但可能会变得一团糟。我修改了一段代码，添加了单独的参数来跟踪parent_key的bool状态，以及携带您想要的字符串的p_key。这是的片段

from pandas import json_normalize
from collections.abc import MutableMapping as mm
def flatten(dictionary, p_key=None, parent_key=False, separator='.'):
items = []
for key, value in dictionary.items():
if parent_key:
new_key = f"{str(p_key)}{separator}{key}"
else:
new_key = p_key if p_key else key
if isinstance(value, mm):
items.extend(flatten(
dictionary=value,
p_key=new_key,
parent_key=True,
separator=separator).items())
elif isinstance(value, list):
for k, v in enumerate(value):
items.extend(flatten(
dictionary={str(k): v},
p_key=new_key,
parent_key=False,
separator=separator).items())
else:
items.append((new_key, value))
return dict(items)
d = { 
"_id" : 1, 
"labelId" : [
6422
], 
"levels" : [
{
"active" : "true", 
"level" : 3, 
"actions" : [
{
"isActive" : "true"
}]
}]
}
x = flatten(d)
x = json_normalize(x)
print(x)