在Python中将数据框转换为字典

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了在Python中将数据框转换为字典，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1245字，纯文字阅读大概需要2分钟。

内容图文

我有一个csv文件,已使用Pandas将其转换为数据框.这是数据框：

Customer ProductID Count

John     1         50
John     2         45
Mary     1         75
Mary     2         10
Mary     5         15

我需要字典形式的输出,如下所示：

{ProductID:1, Count:{John:50, Mary:75}},
{ProductID:2, Count:{John:45, Mary:10}},
{ProductID:5, Count:{John:0, Mary:15}}

我阅读了以下答案：

python pandas dataframe to dictionary
和

Convert dataframe to dictionary

这是我拥有的代码：

df = pd.read_csv('customer.csv') 
dict1 = df.set_index('Customer').T.to_dict('dict') 
dict2 = df.to_dict(orient='records')

这是我当前的输出：

dict1 = {'John': {'Count': 45, 'ProductID': 2}, 'Mary': {'Count': 15, 'ProductID': 5}}

dict2 = [{'Count': 50, 'Customer': 'John', 'ProductID': 1},
 {'Count': 45, 'Customer': 'John', 'ProductID': 2},
 {'Count': 75, 'Customer': 'Mary', 'ProductID': 1},
 {'Count': 10, 'Customer': 'Mary', 'ProductID': 2},
 {'Count': 15, 'Customer': 'Mary', 'ProductID': 5}]

解决方法:

您可以使用IIUC：

d = df.groupby('ProductID').apply(lambda x: dict(zip(x.Customer, x.Count)))
      .reset_index(name='Count')
      .to_dict(orient='records')

print (d)
[{'ProductID': 1, 'Count': {'John': 50, 'Mary': 75}}, 
 {'ProductID': 2, 'Count': {'John': 45, 'Mary': 10}}, 
 {'ProductID': 5, 'Count': {'Mary': 15}}]

内容总结

以上是互联网集市为您收集整理的在Python中将数据框转换为字典全部内容，希望文章能够帮你解决在Python中将数据框转换为字典所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端