如何使用csv文件的2列创建字典?

时间:2019-03-07 13:27:54

标签: python python-3.x pandas dictionary

我有一个csv文件,其数据位于dataframe中:

df = pd.read_csv(r'C:\Users\isha\Desktop\SF head count report 7 mar 2019.csv')

数据框有20多个列,但是我需要仅使用dataframe中的2列来创建字典。

1 个答案:

答案 0 :(得分:1)

这是read_csv中的usecols参数的目的

df = pd.read_csv('my_file.csv', usecols=['first_col','second_col'])

您可以为其提供所需的索引或列名,而Pandas只会将它们存储在df中。

要创建字典,您可以调用df的内置方法df.to_dict()

默认情况下,键将是您的索引,两列是值。如果您只想这两列,一列为键,一列为值,请遵循Chris A的建议:

df.set_index('your_key_column')['your_value_column'].to_dict()