Multiindex GroupBy() 的散点图

Question

我正在尝试使用 Multiindex (http://pandas.pydata.org/pandas-docs/stable/groupby.html#groupby-with-multiindex) 制作 GroupBy() 的散点图。也就是说，我想在 x 轴上绘制一个标签，在 y 轴上绘制另一个标签，将 mean() 绘制为每个点的大小。

df['RMSD'].groupby([df['Sigma'],df['Epsilon']]).mean() 返回：

Sigma_ang  Epsilon_K
3.4        30           0.647000
           40           0.602071
           50           0.619786
3.6        30           0.646538
           40           0.591833
           50           0.607769
3.8        30           0.616833
           40           0.590714
           50           0.578364
Name: RMSD, dtype: float64

我想绘制类似：plt.scatter(x=Sigma, y=Epsilon, s=RMSD)

最好的方法是什么？我无法为每个 RMSD 值获取正确的 Sigma 和 Epsilon 值。

Answer 1

+1 到 Vaishali Garg。根据他的评论，以下工作： df_mean = df['RMSD'].groupby([df['Sigma'],df['Epsilon']]).mean().reset_index() plt.scatter(df_mean['Sigma'], df_mean['Epsilon'], s=100.*df_mean['RMSD'])