Как вписать линию регрессии в диаграмму рассеяния

Question

У меня есть следующие панды, где data.head () показывает таблицу ниже:

          Country     LifeExp   Year        Continent
     0    USA         88        2022       Americas
     1    UK          78        2022       Europe
     2    UAE         78        2022       Asia
     3    Pakistan    98        2022       Asia

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

Я попытался seaborn.scatterplot(x='year',y='lifeExp',hue='continent',data=data) нарисовать необходимый график рассеяния, однако я не знаю, как подогнать линию регрессии к только что построенному графику рассеяния, поскольку у этой функции, кажется, нет опциидля них.

Answer 1

Вы можете использовать seaborn.lmplot(...). Я пробовал это с вашим df.head () в качестве данных (только четыре точки данных).

print(data)
#    Country  LifeExp  Year Continent
#0       USA       88  2022  Americas
#1        UK       78  2022    Europe
#2       UAE       78  2022      Asia
#3  Pakistan       98  2022      Asia

С lmplot,

seaborn.lmplot(x='Year',y='LifeExp',hue='Continent',data=data)

С диаграммой рассеяния

seaborn.scatterplot(x='Year',y='LifeExp',hue='Continent',data=data)