Create dictionary from dataframe without a loop

Question

Below is an example of my dataframe and a dictionary of what I'm looking to compute from it. My current method involves looping through unique ColA values, create a subset dataframe, getting list of ColB values, and make a dict from that. Problem is I have over a million unique ColA values to loop through. Any ideas??

DF

 ColA       ColB
 mike        34
 mike         3
 mike        10
 bill        80
 dean         2
 dean         4
 dean        44
 dean        56

desired dictionary = {'mike':[34,3,10], 'bill': [10], 'dean': [2,4,44,56]}

Any ideas? Thanks!!

Does this answer your question? How to group dataframe rows into list in pandas groupby — Alexandre B.
– Alexandre B., Commented May 12, 2021 at 20:07
I'm assuming the value of 'bill' key should be [80] not [10] — Andrej Kesely
– Andrej Kesely, Commented May 12, 2021 at 20:08

Andrej Kesely · Accepted Answer · 2021-05-12 20:06:58Z

3

out = df.groupby("ColA")["ColB"].agg(list).to_dict()
print(out)

Prints:

{'bill': [80], 'dean': [2, 4, 44, 56], 'mike': [34, 3, 10]}

answered May 12, 2021 at 20:06

Andrej Kesely

196k15 gold badges60 silver badges105 bronze badges

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

Collectives™ on Stack Overflow

Create dictionary from dataframe without a loop

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related