更改从熊猫数据帧生成的json格式

Question

我有一个熊猫数据框，想将其转换为json格式。

import pandas as pd
df = pd.DataFrame({'A': [1,2,3,4], 'B': [5,6,7,8]})
df.to_json()

'{"A":{"0":1,"1":2,"2":3,"3":4},"B":{"0":5,"1":6,"2":7,"3":8}}'

我需要的json字符串如下：

'{"A":[1,2,3,4], "B":[5,6,7,8]}'

如何从pandas数据框中获取此json字符串？非常感谢。

Answer 1

您可以将to_dict()与orient='list'一起使用，并考虑输出的字符串表示形式。

>>> str(df.to_dict(orient='list'))
"{'A': [1, 2, 3, 4], 'B': [5, 6, 7, 8]}"

Answer 2

您是否尝试将参数index = False：

https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.to_json.html

index：bool，默认为True 是否在JSON字符串中包括索引值。仅当Orient为“拆分”或“表格”时，才支持不包括索引（index = False）。

0.23.0版中的新功能。