Create different dataframes from a main dataframe and order by frequency

January 6, 2023

I have a dataframe which looks like this

a     b      name    .....
1     1      abc
2     2      xyz
3     3      abc
4     4      dfg

Now I need to create multiple dataframes based on the frequency of the names like df_abc should have all the data for name "abc" and so on. Tried using for loop but I’m new to python and not able to solve it. Thanks!

df_abc


a    b     name
1    1     abc
3    3     abc

>Solution :

You can use .groupby and list which yields a list of tuples. Using dict-comprehension you can access these dataframes with my_dict["abc"].

df = pd.DataFrame(
    {"a": [1, 2, 3, 4], "b": [1, 2, 3, 4], "name": ["abc", "xyz", "abc", "dfg"]}
)
my_dict={name:df for name, df in list(df.groupby("name")) }

for val, df_val in my_dict.items():
    print(f"df:{df_val}\n")