Create multiple dataframe using for loop

Question

Hi I have a dataset looking like this.

https://i.sstatic.net/cjYAW.png

I want to create mutiple dataframes using countryname_df = data.loc[data.location == 'country name'. This is an example result:

https://i.sstatic.net/UHkQ2.png

May I know if I can use loop for to create with this format:

     c_df = data.loc[data.location == c]

So you want to split the dataframe into horizontal slices , rows? Or only for a list of specific countries? — roadrunner66
– roadrunner66, Commented Sep 25, 2020 at 2:51
Does this answer your question? Create Pandas DataFrames from Unique Values in one Column — Trenton McKinney
– Trenton McKinney, Commented Sep 25, 2020 at 3:13
Could you make a minimal example where you show a specific dataframe as input and what you want your output to look like? — roadrunner66
– roadrunner66, Commented Sep 25, 2020 at 20:13

BP34500 · Accepted Answer · 2020-09-25 03:08:45Z

1

the following example does what you want.

start by creating an empty list where you will store all the different dataframes.

Get a list of the unique countries.

import pandas as pd

data = {'country':['USA','USA','CANADA','CANADA','CANADA','SPAIN','SPAIN','PERU','PERU','PERU','PERU','PERU'],
        'col_1': [3, 2, 1, 0,235,2,5,7,9,7,14,346], 
        'col_2': ['a', 'b', 'c', 'd','v','asd','sg','sdg','ery','wqrew','asf','Ùùsd'],
        'col_3':[3234,52345,64534,65652,1234,435,346,7687,969,689689,79,2143]}
df = pd.DataFrame.from_dict(data)



list_of_df = []
unique_countries = set(list(df['country']))
for country in unique_countries:
  list_of_df.append(df.loc[df['country'] == country,:])

# this is the first dataframe of the list.
list_of_df[0]

answered Sep 25, 2020 at 3:08

BP34500

1789 bronze badges

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

Comments

roadrunner66 · Accepted Answer · 2020-09-25 03:04:47Z

0

import pandas as pd

# a minimal example often helps
d = {'country': ['Albania','Bahrain','Congo'], 'sales': [10,20,30]}
df = pd.DataFrame(data=d)

countries=['Albania','Congo']


for c in countries:
     print(df[df['country'] == c])

output:

country  sales
0  Albania     10
country  sales
2   Congo     30

answered Sep 25, 2020 at 3:04

roadrunner66

7,9914 gold badges34 silver badges39 bronze badges

Collectives™ on Stack Overflow

Create multiple dataframe using for loop

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related