将pandas.DataFrame转换为Python中的字典列表

Question

我有一个字典,从数据帧转换如下:

a = d.to_json(orient='index')

字典:

{"0":{"yr":2017,"PKID":"58306, 57011","Subject":"ABC","ID":"T001"},"1":{"yr":2018,"PKID":"1234,54321","Subject":"XYZ","ID":"T002"}}

我需要的是它在列表中,所以基本上是一个字典列表.所以我只添加一个[],因为这是在其余代码中使用的格式.

input_dict = [a]

input_dict:

['
{"0":{"yr":2017,"PKID":"58306, 57011","Subject":"ABC","ID":"T001"},"1":{"yr":2018,"PKID":"1234,54321","Subject":"XYZ","ID":"T002"}}
']

我需要在[和之前]之后删除单引号.此外,以列表的形式具有PKID值.

怎么能实现这一目标？

预期产出:

[ {"yr":2017,"PKID":[58306, 57011],"Subject":"ABC","ID":"T001"},"1":{"yr":2018,"PKID":[1234,54321],"Subject":"XYZ","ID":"T002"} ]

注意:PKID列有多个整数值,必须作为整数的提升.字符串是不可接受的.所以我们需要像"PKID":[58306,57011]而不是"PKID":"[58306,57011]"

Answer 1

pandas.DataFrame.to_json返回一个字符串(JSON字符串),而不是字典.尝试to_dict改为:

>>> df
   col1  col2
0     1     3
1     2     4
>>> [df.to_dict(orient='index')]
[{0: {'col1': 1, 'col2': 3}, 1: {'col1': 2, 'col2': 4}}]
>>> df.to_dict(orient='records')
[{'col1': 1, 'col2': 3}, {'col1': 2, 'col2': 4}]