Фрейм данных Big pandas для определения некоторых столбцов

#python #pandas #dataframe #dictionary

#python #pandas #фрейм данных #словарь

Вопрос:

У меня есть большой фрейм данных, например:

        product   price    serial     category    department   origin    
 0     cookies       4      2345    breakfast          food        V
 1     paper       0.5      4556   stationery          work        V
 2     spoon         2      9843      kitchen     household        M
  

Я хочу преобразовать в dict, но мне просто нужен вывод, подобный:

    {serial: 2345}{serial: 4556}{serial: 9843} and {origin: V}{origin: V}{origin: M}
  

где ключ — это имя столбца, а значение — значение

Теперь я попробовал с помощью df.to_dict(‘values’) и выбрал dic[‘origin’] и возвращает меня

    {0: V}{1:V}{2:M}
  

Я тоже пробовал с df.to_dict(‘records’), но это дает мне:

    {product: cookies, price: 4, serial:2345, category: breakfast, department:food, origin:V}
  

и я не знаю, как выбрать только «источник» или «серийный номер»

Ответ №1:

Вы можете сделать что-то вроде:

 serial_dict = df[['serial']].to_dict('r')
origin_dict = df[['origin']].to_dict('r')