熊猫：CSV到字典

Question

我有以下格式的csv文件：

  

id，类别

     

1，苹果

     

2，橙色

     

3，香蕉

我需要阅读此文件并填充一个词典，该词典的ID为键，类别为值。

我正在尝试使用panda，但它的to_dict函数使我们返回了具有单个键值对的字典。键是另一个包含1,2,3作为条目的字典，值也是一个包含苹果，橙子，香蕉的字典。

我想要的是三个键值条目，例如1->苹果，2->橙子，3->香蕉。

Answer 1

您可以按第一列创建索引，然后将df['category']系列转换为字典：

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

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

或者如果仅2列csv可用，则通过index_col=0和squeeze=True参数创建Series并转换为字典：

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()