1. ホーム
  2. python

Python で散布図に直線をオーバープロットするには?

2023-09-13 13:43:25

質問

2つのデータのベクトルがあり、それらを matplotlib.scatter() . 今、私はこれらのデータへの線形フィットをオーバープロットしたいと思います。どのようにすればよいのでしょうか?私は scikitlearnnp.scatter .

どのように解決するのですか?

import numpy as np
from numpy.polynomial.polynomial import polyfit
import matplotlib.pyplot as plt

# Sample data
x = np.arange(10)
y = 5 * x + 10

# Fit with polyfit
b, m = polyfit(x, y, 1)

plt.plot(x, y, '.')
plt.plot(x, b + m * x, '-')
plt.show()

<イグ