1. 程式人生 > 程式設計 >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

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。