Python 线性回归可视化 并将回归函数放置到图像上
import matplotlib.pyplot as plt
import scipy
import seaborn as sns# 加载内置的数据集
df = sns.load_dataset('tips')#create regplot
p = sns.regplot(x='total_bill', y='tip', data=df)#calculate slope and intercept of regression equation
slope, intercept, r, p, sterr = scipy.stats.linregress(x=p.get_lines()[0].get_xdata(),y=p.get_lines()[0].get_ydata())#add regression equation to plot
result_string = f"$y = {slope:.2f}x + {intercept:.2f}$"plt.figtext(0.5, 0.6, result_string)plt.show()运行结果如下：
2024年2月7日22:33:05