python - 将 Pandas 数据框转换为以第一列为键的字典

Question

我有一个 Pandas 数据框:

A  ||  B  ||  C
x1    x      [x,y]
x2    a      [b,c,d] 

我正在尝试制作一个看起来像这样的字典:

{x1: {B : x, c : [x,y]}, x2: {B: a, C:[b,c,d}}

我试过 to_dict函数，但这会将整个数据框更改为字典。我对如何迭代第一列并使其成为键而将 df 的其余部分设为字典作为该键的值有点迷茫。

Answer 1

尝试:

x = df.set_index("A").to_dict("index")
print(x)

打印:

{'x1': {'B': 'x', 'C': ['x', 'y']}, 'x2': {'B': 'a', 'C': ['b', 'c', 'd']}}