Pandas dummies to categorical trick

import pandas as pd
df = sns.load_dataset('exercise',index_col=0)
dummies = pd.get_dummies(df['diet'])
dummies.idxmax(axis=1)

Being able to convert a matrix of dummy variables into a single categorical column comes in handy in a lot of situations

Tweet Share