Create multiple data frames from one -python

Question

I have a large file (>500 rows) with multiple data points for each unique item in the list, something like :

cheese	weight	location
gouda	1.4	AL
gouda	2	TX
gouda	1.2	CA
cheddar	5.3	AL
cheddar	6	MN
chaddar	2	WA
Havarti	4	CA
Havarti	4.2	AL

I want to make data frames for each cheese to store the relevant data

I have this:

main_cheese_file = pd.read_csv('CheeseMaster.csv')

cut_the_cheese = main_cheese_file.cheese.unique()
melted = {elem: pd.DataFrame() for elem in cut_the_cheese}

for slice in melted.slice():
    melted[slice] = main_cheese_file[:][main_cheese_file.cheese == slice]

to split it up on the unique thing I want.

What I want to do with it is make df's that can be exported for each cheese with the cheese name as the file name.

So far I can force it with

 melted['Cheddar'].to_csv('Cheddar.csv')

and get the Cheddars ....

but I don't want to have to know and type out each type of cheese on the list of 500 rows...

Is there a way to add this to my loop?

It_is_Chris · Accepted Answer · 2022-01-12 16:04:49Z

1

You can just iterate over a groupby object

import pandas as pd

df = pd.read_csv('CheeseMaster.csv')
for k,v in df.groupby('cheese'):
    v.to_csv(f'{k}.csv', index=False)

edited Jan 12, 2022 at 16:04

answered Jan 12, 2022 at 15:58

It_is_Chris

14.2k3 gold badges27 silver badges45 bronze badges

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

4 Comments

Porsche Adams Over a year ago

thanks! This does look a lot simpler - It only split out one csv file called {k} though ...

It_is_Chris Over a year ago

@PorscheAdams make sure to use the f-string: f'{k}.csv'

Porsche Adams Over a year ago

well - that makes a difference! thank you! and thanks for helping me edit the question - I could not figure out how to make a nice table for my cheeses!

It_is_Chris Over a year ago

@PorscheAdams No problem and good luck with all that cheese.

Collectives™ on Stack Overflow

Create multiple data frames from one -python

1 Answer 1

4 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

4 Comments

Your Answer

Sign up or log in

Post as a guest

Related