python：从 dict 到 df 再回到 dict

Question

假设我们有这个字典

d = {'A':True, 'B':False, 'C':True}

使用此命令，我得到了所需的 Dataframe：

import pandas as pd
df = pd.DataFrame(d.items())

   0      1
0  A   True
1  B  False
2  C   True

但是当我想转换回字典时，我得到了这个：

d_again = df.to_dict('records')

[{0: 'A', 1: True}, {0: 'B', 1: False}, {0: 'C', 1: True}]

这与第一个 dict d 不同。

谢谢您的帮助： ：）

Answer 1

您可以使用 dataframe 的values属性并将其传递给dict

d_again = dict(df.values)

{'A': True, 'B': False, 'C': True}

Answer 2

直接翻译是

print(df.set_index(0)[1].to_dict())

哪个产量

{'A': True, 'B': False, 'C': True}

完整代码：

import pandas as pd

dct_in = {'A':True, 'B':False, 'C':True}
df = pd.DataFrame(dct_in.items())
dct_out = df.set_index(0)[1].to_dict()
assert dct_in == dct_out