Построение pandas DataFrame - group by - как добавить разные заголовки к каждому произведенному графику

Question

У меня есть pandas фрейм данных, и я хочу построить одно значение относительно другого, основываясь на определенном поле. Итак, у меня есть 5 разных типов в 'Topi c', и я хочу построить каждый. Код как ниже в настоящее время.

    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 к каждому из них, который совпадает с тем из Topi c? Так что, если topi c был «Астма», я хочу, чтобы заголовок / x был «Астма», а затем следующий - «Бронхит» и так далее. 2. Я хочу добавить их на том же графике, если это возможно, я буду решать, сколько из них выглядит хорошо, когда я их вижу. Как мне это сделать? 3. (Бонусный вопрос !!) Могу ли я легко добавить линию «наилучшего соответствия» к каждому сюжету?

Спасибо всем.

Answer 1

groupby.plot метод возвращает список осей, которые можно использовать для установки заголовков, например, если есть только две оси:

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

titles = ['Asthma', 'Bronchitis']
count = 0
for ax in axes:
    ax.set_title(f'{titles[count]}')
    ax.set_xlabel(f'{titles[count]}')
    count += 1

plt.show()

Для наилучшего соответствия я предполагаю, что это линейная регрессия, так что стоит проверить regplot