Матрица разброса с линией тренда и 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, а также добавить линию тренда на каждый из графиков. Мне было интересно, есть ли способ сделать это.

0 ответов

Другие вопросы по тегам