绘制pandas DataFrame-分组依据-如何为每个生成的图添加不同的标题

Question

我有一个熊猫数据框，并希望根据特定字段绘制一个值与另一个值的关系。所以我在“主题”中有5种不同的类型，我想分别绘制。目前代码如下。

    dfCombinedToPlot.groupby('Topic').plot(x='DataValue', y='CasesPer100kPop', style='o')  
    # plt.title()  Want this to equal "'Topic' vs number of cases"
    # plt.xlabel() Want this to equal 'Topic'  
    # plt.ylabel()  
    plt.show()

我有3个问题。1.我是否可以在每个与标题相匹配的标题/ xlabel中添加标题/ xlabel？因此，如果主题是“哮喘”，我希望标题/ x标签为“哮喘”，然后下一个为“支气管炎”，依此类推。2.如果可能，我想将它们添加到同一图上，当我看到它们时，我将决定有多少看起来不错。我该怎么做呢？3.（奖金问题！！）我可以轻松地在每个图上添加一条“最佳拟合”线吗？

谢谢。

Answer 1

groupby.plot方法返回可用于设置标题的轴列表，例如：

axes = dfCombinedToPlot.groupby('Topic').plot(x='DataValue', y='CasesPer100kPop', style='o')

count = 0
for ax in axes:
    ax.set_title(f'title {count}')
    count += 1

plt.show()