python - Pandas DataFrame 将某些列合并到嵌套 JSON 中

Question

假设您有一个如下所示的 pandas DataFrame:

A1a  A1b   A2a    A2b   …   B1a …
0.25 0.75  0.10   0.5       1   
…    …     …      …         …   

并且您想要输出一个 JSON 对象列表(每一行一个对象)，如下所示:

[
    {
        A: {
            1: {
                a: 0.25,
                b: 0.75
            },
            2: {
                a: 0.1,
                b: 0.5,
                ...
            },
            ...
        },
        B: {
            1: {
                a: 1
            },
            ...
        },
        ...
    },
    ...
]

最好的方法是什么？

这里显然有很多 pandas/nested JSON 问题，但我认为这是不同的，因为我试图将特定列嵌套在同一行中，而不是对列中具有相同值的行进行分组(例如在 this example 中)。

Answer 1

既然你链接了这个页面，我就借用recur_dictify来自该链接中接受的答案的函数

#make your df columns become multiple index 
df.columns=pd.MultiIndex.from_tuples(df.columns.map(list).map(tuple))

      A
      1          2
      a     b    a    b
0  0.25  0.75  0.1  0.5

#Then we apply the function
recur_dictify(df.T.reset_index())

{'A': {'1': {'a': 0.25, 'b': 0.75}, '2': {'a': 0.1, 'b': 0.5}}}