我有一个csv
文件,其数据位于dataframe
中:
df = pd.read_csv(r'C:\Users\isha\Desktop\SF head count report 7 mar 2019.csv')
数据框有20多个列,但是我需要仅使用dataframe
中的2列来创建字典。
答案 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()