python中如何实现DataFrame转dict字典

这篇文章主要为大家展示了“python中如何实现DataFrame转dict字典”，内容简而易懂，条理清晰，希望能够帮助大家解决疑惑，下面让小编带领大家一起研究并学习一下“python中如何实现DataFrame转dict字典”这篇文章吧。

背景：将商品id以及商品类别作为字典的键值映射，生成字典，原为DataFrame

# 创建一个DataFrame
# 列值类型均为int型
import pandas as pd
item = pd.DataFrame({'item_id': [100120, 10024504, 1055460], 'item_category': [87974, 975646, 87974]}, index=[0, 1, 2])
item

# 将item_id,item_category两列数值转为dict字典
# 注意：同种商品类别肯定会对应不同商品，即一对多，进行字典映射，一定要是item_id作为键，item_category作为值
# 由于原始数据为int类型，结果将是字符串之间的映射，因此需要对列值进行数据类型转换
item.item_id = (item['item_id']).astype(str)
item.item_category = (item['item_category']).astype(str)
item_dict = item.set_index('item_id')['item_category'].to_dict()
item_dict

以上是“python中如何实现DataFrame转dict字典”这篇文章的所有内容，感谢各位的阅读！相信大家都有了一定的了解，希望分享的内容对大家有所帮助，如果还想学习更多知识，欢迎关注亿速云行业资讯频道！