pandas 写入字典_python - 将Pandas DataFrame转换为字典

to_dict()方法将列名称设置为字典键，因此您需要稍微重新整形DataFrame。 设置＆＃39; ID＆＃39; 列作为索引，然后转置DataFrame是实现此目的的一种方法。

to_dict()也接受“东方”的定位。 您为了输出每列的值列表而需要的参数。 否则，将为每列返回表单{index: value}的字典。

可以使用以下行完成这些步骤：

>>> df.set_index('ID').T.to_dict('list')

{'p': [1, 3, 2], 'q': [4, 3, 2], 'r': [4, 0, 9]}

如果需要不同的字典格式，这里是可能的东方参数的示例。 考虑以下简单的DataFrame：

>>> df = pd.DataFrame({'a': ['red', 'yellow', 'blue'], 'b': [0.5, 0.25, 0.125]})

>>> df

a b

0 red 0.500

1 yellow 0.250

2 blue 0.125

然后选项如下。

dict - 默认值：列名是键，值是索引的字典：数据对

>>> df.to_dict('dict')

{'a': {0: 'red', 1: 'yellow', 2: 'blue'},

'b': {0: 0.5, 1: