熊貓：csv 到字典

Question

我有一個 csv 文件，格式如下：

編號，類別

1、蘋果

2、橙色

3、香蕉

我需要閱讀這個文件並填充一個字典，其中 id 作為鍵，類別作為值。

我正在嘗試使用熊貓，但它的 to_dict function 我們返回一個帶有單個鍵值對的字典。 Key 是另一個包含 1,2,3 作為條目的字典，而 value 也是一個包含 apple、orange、banana 的字典。

我想要的是三個鍵值條目，例如 1 -> apple、2 -> ornge、3 ->banana。

Answer 1

您可以按第一列創建索引，然后將 Series df['category']轉換為字典：

df = pd.read_csv('file.csv', index_col=0)

d = df['category'].to_dict()

或者，如果只有 2 列 csv 可以通過index_col=0和squeeze=True參數創建系列並轉換為字典：

d = pd.read_csv('file.csv', index_col=0, squeeze=True).to_dict()

或者，如果更多列並且不需要按第一列使用索引：

df = pd.read_csv('file.csv')

d = df.set_index('id')['category'].to_dict()