python - Loop Through Pandas Dataframe and split into multiple dataframes based on unique column values

Question

I have a dataframe saved in a list. Is there a way to loop through the list to create separate dataframes based of a column value?

ex: Turn this

df

ID	Colour	Transport
0902	red	car
0902	blue	car
0105	red	car
0105	yellow	car
0105	orange	boat

Answer 1

Try this:

df1 = df.loc[df['ID'] == 0902]
df2 = df.loc[df['ID'] == 0105]

Or this:

df1, df2 = [group for _, group in df.groupby('ID')]

Or if you want it dynamically:

dct = {f'df{idx}': group for _, group in df.groupby('ID')]}
print(dct)

Or:

dct = {}
for idx, v in enumerate(df['ID'].unique()):
    dct[f'df{idx}'] = df.loc[df['ID'] == v]

print(dct)

And print like this for specific dataframe:

print(dct['df1'])