熊猫df至python嵌套字典

Question

假设我下面有df：

df = pd.DataFrame({
    'A': ['a', 'b'],
    'B': [1, 2],
    'C': [3, 4]
})

如何将其转换为以下格式的嵌套dict：

{'a': {'B': 1, 'C': 2}, 'b': {'B': 3, 'C': 4}}

我尝试过df.to_dict()，但不会返回我想要的格式。

谢谢！

Answer 1

尝试set_index

df.set_index('A').to_dict('index')
Out[8]: {'a': {'B': 1, 'C': 3}, 'b': {'B': 2, 'C': 4}}

Answer 2

您可以将A设置为索引并使用to_dict：

df.set_index('A').to_dict('i')
# {'a': {'B': 1, 'C': 3}, 'b': {'B': 2, 'C': 4}}