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