python - 如何使用 2 列 csv 文件创建字典？

Question

我有一个csv文件，它的数据在dataframe：

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

数据框有 20 多列，但我需要仅使用 2 列来创建字典dataframe。

标签： pythonpython-3.xpandasdictionary

Answer 1

这是read_csvusecols中参数的用途

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

你可以给它你想要的列的索引或名称，Pandas 只会将这些存储在你的 df.xml 中。

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

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

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