python - 获取适当的标签以显示在分组的 DataFrameGroupBy.plot() 图例中

Question

我想知道如何在使用该group by方法分组后让正确的图例标签显示在 DataFrame 图中。人们会期望图例标签是组名；相反，它们显示为冗余的因变量名称。

例如，我有一个如下所示的 DataFrame：

     ECMO  Year Sex  Runs    p_ecmo
102   111  2011   M  2106  0.052707
104    31  2012   F  1801  0.017213
105    42  2012   M  2664  0.015766
107    59  2013   F  1039  0.056785
108    72  2013   M  1386  0.051948

我正在尝试按年份分组sex和绘图p_ecmo。因此，

ecmo_by_year[ecmo_by_year.Year>=1996].groupby('Sex').plot('Year', 'p_ecmo', grid=False)
ylabel('Proportion ECMO')
plt.legend()

但我得到的是这个，这没有帮助。图例应显示组名，而不是变量。

Answer 1

它看起来不像您需要groupby的，因为您没有在 p_ecmo 上拆分、合并或应用

您可以使用 apivot_table来获得所需的结果。

pt = ecmo_by_year.pivot_table(['p_ecmo'], rows='Year', cols='Sex').fillna(0) pt['p_ecmo'].plot()