CSV to dictionary having headers as keys and the rest of rows as lists of strings (values) in Python

Question

I have a CSV like this:

(row 0)    key1   key2   key3
(row 1)    A      D      G
(row 2)    B      E      H
(row 3)    C             I

I want to read it and get the information into a dictionary as follows: dict = {key1: ["A", "B", "C"], key2: ["D", "E"], key3: ["G", "H", "I"]}.

So basically I want to have the first row as keys and all the values per column as values of a list (the list is the value of the key in the dictionary).

Thank you!

yatu · Accepted Answer · 2020-10-05 07:25:58Z

1

You could stack and groupby aggregating the values to a list, and turn to a dictionary with .to_dict():

df = pd.read_csv("my_csv.csv")
df.stack().groupby(level=1).agg(list).to_dict()
# {'key1': ['A', 'B', 'C'], 'key2': ['D', 'E'], 'key3': ['G', 'H', 'I']}

edited Oct 5, 2020 at 7:25

answered Oct 5, 2020 at 7:23

yatu

88.7k12 gold badges93 silver badges148 bronze badges

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

2 Comments

oreopot Over a year ago

in question, I is in key3 not in key2

yatu Over a year ago

Yes, that's just for how I read the string @oreopot Thanks

Collectives™ on Stack Overflow

CSV to dictionary having headers as keys and the rest of rows as lists of strings (values) in Python

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related