python - 数据框列到字典的字典中

Question

我有一个像这样的数据框:

subject   score_A   score_B   score_C
math        90        75        50
art         85        65        45
econ        90        80        60

我想把它们变成像这样的字典:

{
   'math':{'A': 90,
           'B': 75,
           'C': 50},
   'art': {'A': 85,
           'B': 65,
           'C': 45},
   'econ':{'A': 90,
           'B': 80,
           'C': 60},
}

实际的数据框更长更宽(每个主题的键比 A、B 和 C 多)。

我想这与生成 A、B、C 等较低级别键的列表以及使用 zip 函数有关，但我有限的尝试都没有奏效。

Answer 1

您可以将subject设置为索引，然后使用带有orient参数的to_dict -

df.set_index('subject').to_dict(orient='index')