python – 将Pandas DataFrame转换为字典

我有一个DataFrame有四列。我想把这个DataFrame转换成一个python字典。我想要第一列的元素是键,同一行中的其他列的元素是值。

数据帧:

    ID   A   B   C
0   p    1   3   2
1   q    4   3   2
2   r    4   0   9  

输出应该是这样的:

字典:

{'p': [1,3,2], 'q': [4,3,2], 'r': [4,0,9]}
to_dict()方法将列名设置为字典键,因此您需要稍微重塑DataFrame。将“ID”列设置为索引,然后设置DataFrame是实现此目的的一种方法。

to_dict()还接受一个outtype关键字参数,您需要为其输出每列的值列表。否则,将为每列返回一个{index:value}形式的字典。

这些步骤可以通过以下行来完成:

>>> df.set_index('ID').T.to_dict('list')
{'p': [1, 3, 2], 'q': [4, 3, 2], 'r': [4, 0, 9]}
http://stackoverflow.com/questions/26716616/convert-a-pandas-dataframe-to-a-dictionary

本站文章除注明转载外,均为本站原创或编译
转载请明显位置注明出处:python – 将Pandas DataFrame转换为字典