python - DataFrame 到 Json，使用第一个 Col 作为键，第二个作为值

Question

我有一个看起来像这样的数据框：

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

我想插入一个如下所示的 mongoDB：

dic = {1:'a', 2:'b',3:'c',4:'d'}

我怎么能做到？

我已经检查过这样的事情，但它们似乎不适用于我的 df：

将 pandas 数据帧转换为 json 对象 - pandas

提前致谢！

标签： pythonjsonmongodbpandasdataframe

Answer 1

使用Series.to_json并在必要时更改key值添加rename：

print (k.set_index('A').rename(columns={'B':'index1'}).to_json())
{"index1":{"1":"a","2":"b","3":"c","4":"d"}}

如果需要导出到文件：

k.set_index('A').rename(columns={'B':'index1'}).to_json('file.json')