Create dictionary from dataframe using multiple columns

Question

I need to create a dictionary where each element is a row. The key represents a specific column and the values will be a list of the remaining column entries. Simple example below:

example df:

Col_A     Col_B    Col_C    Col_D
Roger      5        blue    house
Kim        2        green   car
Mike       10       red     bike

desired dictionary = {'Roger':['5','blue','house'],'Kim':['2','green','car'],'Mike':['10','red','bike']}

Any suggestions are appreciated!

Shubham Sharma · Accepted Answer · 2021-06-08 18:25:56Z

4

Set the index to COl_A then transpose the frame and convert to dictionary with orientation set to list

df.set_index('Col_A').T.to_dict('list')

{'Roger': [5, 'blue', 'house'],
 'Kim': [2, 'green', 'car'],
 'Mike': [10, 'red', 'bike']}

answered Jun 8, 2021 at 18:25

Shubham Sharma

71.8k6 gold badges26 silver badges58 bronze badges

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

Comments

Andrej Kesely · Accepted Answer · 2021-06-08 18:22:51Z

2

Try:

print(df.set_index("Col_A").apply(list, 1).to_dict())

Prints:

{'Roger': [5, 'blue', 'house'], 'Kim': [2, 'green', 'car'], 'Mike': [10, 'red', 'bike']}

answered Jun 8, 2021 at 18:22

Andrej Kesely

196k15 gold badges60 silver badges105 bronze badges

Collectives™ on Stack Overflow

Create dictionary from dataframe using multiple columns

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related