Матрица разброса с линией тренда и r-квадрат / коэффициент корреляции в python
Я пытаюсь построить матрицу рассеяния с линиями тренда и r-sqaured в качестве легенды для каждого из графиков рассеяния. Скажем, для следующего фрейма данных:
df = pd.DataFrame(
{'x':[2.1,3.1,5.4,8,12,11],
'y':[3,5.3,4,6,8,9.1],
'c':[7,4,5,2,3,1]
}
)
Матрица разброса панд дает мне:
pd.plotting.scatter_matrix(df,marker='o')
и я могу найти коэффициент детерминации (r2), используя тепловую карту seaborn:
import seaborn as sns
corr = df.corr()**2
sns.heatmap(corr,square=True,annot=True)
теперь я хочу показать все эти r2 как легенды на первой scatter_matrix, а также добавить линию тренда на каждый из графиков. Мне было интересно, есть ли способ сделать это.