Постройте отдельные точки данных в каждой группе после pandas groupby

Question

Я хотел бы сгруппировать фрейм данных, используя несколько критериев, а затем визуализировать отдельные точки данных в каждой группе, используя разбросанный график.

import pandas as pd
import seaborn as sns

df_tips = sns.load_dataset('tips')
df_tips.groupby(['sex', 'day', 'smoker'])['tip']    # How could I scatter plot individual tip in each group?

В идеале хотелось бы, чтобы было что-то вроде этого:

Answer 1

Я бы сделал:

df_tips = sns.load_dataset('tips')
groups = df_tips.groupby(['sex', 'day', 'smoker'])['tip']

fig,ax = plt.subplots()
for  i,(k,v) in enumerate(groups):
    ax.scatter([i]*len(v), v)

ax.set_xticks(np.arange(len(groups)))
ax.set_xticklabels([k for k,v in groups],rotation=90);

Вывод:

Answer 2

Я нашел более простой способ сделать это, и сюжет более красивый (я думаю).

import pandas as pd
import seaborn as sns

df_tips = sns.load_dataset('tips')
df_tips['Groups'] = df_tips[['sex', 'day', 'smoker']].astype(str).agg('.'.join, axis=1)
sns.swarmplot(x='Groups', y='tip', data=df_tips)
plt.xticks(
    rotation=90, 
    fontweight='light',
    fontsize='x-large'  
)

Вот результат: