My goal in to split python dataframe by multiple columns. In the case of one column, data frame can be splitted by column ‘X1’ as below, using the groupby method. However, how to split dataframe according to columns X1 and X2?
df = pd.DataFrame({'X1': ['Falcon', 'Falcon', 'Parrot', 'Parrot'],
'X2': ['Captive', 'Wild', 'Captive', 'Wild'],
'X3': ['BIG', 'SMALL', 'BIG', 'SMALL']})
dfs= dict(tuple(df.groupby('X1')))
dfs
Advertisement
Answer
If pass another column name is necessary select dfs by tuples:
dfs= dict(tuple(df.groupby(['X1', 'X2'])))
print (dfs[('Falcon','Captive')])
X1 X2 X3
0 Falcon Captive BIG
If want select by strings is possible use join in dict comprehension:
dfs={f'{"_".join(k)}' : v for k, v in df.groupby(['X1', 'X2'])}
print (dfs['Falcon_Captive'])
X1 X2 X3
0 Falcon Captive BIG