draw regression lines with axline

baogianghoangvu · baogianghoangvu · commit 1531619e7d21 · 2020-10-26T22:22:17.000+07:00
diff --git a/examples/specialty_plots/anscombe.py b/examples/specialty_plots/anscombe.py
@@ -40,10 +40,8 @@
     ax.plot(x, y, 'o')
 
     # linear regression
-    p1, p0 = np.polyfit(x, y, deg=1)
-    x_lin = np.array([0, 20])
-    y_lin = p1 * x_lin + p0
-    ax.plot(x_lin, y_lin, 'r-', lw=2)
+    p1, p0 = np.polyfit(x, y, deg=1)  # slope, intercept
+    ax.axline(xy1=(0, p0), slope=p1, color='r', lw=2)
 
     # add text box for the statistics
     stats = (f'$\\mu$ = {np.mean(y):.2f}\n'
