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