Блог

Главная - Вопросы по программированию - Создание словаря на основе столбцов фрейма данных

Создание словаря на основе столбцов фрейма данных

#python #python-3.x #pandas #dictionary

#python #python-3.x #pandas #словарь

Вопрос:

У меня есть фрейм данных вида:

 pandas as pd
df = pd.DataFrame({'col1': [1,2,3,4,5,6], 
                   'col2': ['A','A','B','C','D','D']}, 
                   index=['row1', 'row2','row3', 'row4','row5', 'row6'])

Я хотел бы создать следующий словарь на основе значений фрейма данных df:

 dict = {'A':[1,2], 'B':[3], 'C':[4], 'D':[5,6]}

Ответ №1:

Преобразуйте значения col1 для групп по GroupBy.apply в списки, а затем вызовите Series.to_dict :

 d = df.groupby('col2')['col1'].apply(list).to_dict()
print (d)
{'A': [1, 2], 'B': [3], 'C': [4], 'D': [5, 6]}