Dataframe to dictionary Python

Question

I'm looking to turn two columns in a dataframe into a dictionary, column_a holds the item, and column_b has the description:

column_a     column_b
---------------------
Furniture    Table
Furniture    Chair
Tech         Laptop
Tech         Screen

And the result should be:

{'Furniture':['Table','Chair'], 'Tech' : ['Laptop', 'Screen']}

Audrey · Accepted Answer · 2021-06-23 12:57:52Z

1

You'll want to use a groupby statement, like this:

def turn_df_into_dict(df):
    
    output_dict = {}
    
    for name, group in df.groupby('column_a'):
        output_dict[name] = group['column_b']
        
    return output_dict

This should give you the intended output :)

answered Jun 23, 2021 at 12:57

Audrey

714 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Daniel Over a year ago

Yep, that's good, however, it does not deal with repeated values in 'column_b' so I added a .unique() :) thanks

oskros · Accepted Answer · 2021-06-23 12:56:15Z

1

Group into a list and then convert to dict:

dct = df.groupby('column_a')['column_b'].apply(list).to_dict()

answered Jun 23, 2021 at 12:56

oskros

3,3452 gold badges13 silver badges36 bronze badges

Collectives™ on Stack Overflow

Dataframe to dictionary Python

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related