python - 如何在 Matplotlib 中添加基于斜率和截距的线？

Question

在 R 中，有一个名为 abline 的函数，其中可以根据截距(第一个参数)和斜率(第二个参数)的规范在绘图上绘制一条线。例如，

plot(1:10, 1:10)
abline(0, 1)

截距为 0 且斜率为 1 的线跨越整个绘图范围。 Matplotlib中有这样的功能吗？

Answer 1

其中许多解决方案都专注于在图中添加一条适合数据的线。这是基于斜率和截距向绘图添加任意线的简单解决方案。

import matplotlib.pyplot as plt 
import numpy as np    

def abline(slope, intercept):
    """Plot a line from slope and intercept"""
    axes = plt.gca()
    x_vals = np.array(axes.get_xlim())
    y_vals = intercept + slope * x_vals
    plt.plot(x_vals, y_vals, '--')