python 迴歸

import numpy as np

import matplotlib.pyplot as plt

#自訂x,y

x = np.array([1,2,3,4,5])

y = np.array([2,5,7,8,10])

#==========

# 散佈圖 + 迴歸

fig, ax = plt.subplots(figsize=(8, 4))

ax.scatter(x, y)

m, c = np.polyfit(x, y, 1) # 使用 Numpy 的 polyfit，參數 1 代表一維，算出 fit 直線斜率

ax.plot(x, m * x + c) # 使用 y = m * x + c 斜率和常數匯出直線

fig.suptitle('Scatter with regression')

plt.show()

#==========

predict = lambda x: m * x + c

print(predict(5))