Блог

У меня есть csv файл, его данные в dataframe :

 df = pd.read_csv(r'C:UsersishaDesktopSF head count report 7 mar 2019.csv')
  

Фрейм данных содержит более 20 столбцов, но мне нужно создать словарь, используя только 2 столбца из dataframe .

Это назначение usecols параметра в read_csv

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

Вы можете присвоить ему индексы или названия столбцов, которые вы хотите, и Pandas сохранит только их в вашем df.

Чтобы создать словарь, вы можете вызвать встроенный метод df df.to_dict()

По умолчанию ключом будет ваш индекс, а двумя столбцами — значения. Если вам нужны просто эти 2 столбца, один в качестве ключа, а другой в качестве значения, тогда следуйте совету Криса А.:

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