Python中的R abline()等效项

import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots()
fig.set_size_inches(8, 6)
ax.set_title('Plot with abline')

x = np.array([1, 2, 3, 4, 5])
y = np.array([2, 4, 6, 8, 10])
ax.scatter(x, y, color='blue', label='Data points')

slope = 2
intercept = 0

def abline(slope, intercept):
    """Compute y values for a given slope and intercept."""
    return slope * x + intercept

ax.plot(x, abline(slope, intercept), color='red', label='abline')

ax.legend()
ax.grid(True)

plt.show()