python 一阶线性拟合

原文链接: python 一阶线性拟合

上一篇: python html 转 pdf

下一篇: sympy 符号函数求导

使用python 对数据进行一阶线性拟合，并获取拟合参数

画出拟合的图像，与所给点进行比较

import numpy as np
import random
import matplotlib.pyplot as plt

x = [i for i in range(10)]
y = [2 * i + 1 + random.random() for i in x]
# 一维线性拟合
f = np.polyfit(x, y, 1)
print(f)
plt.plot(x, y, 'ro')
x = np.linspace(0, 10, 1000)
y = f[0] * x + f[1]
plt.plot(x, y)
plt.show()

由于random模块产生0--1之间的随机数，所以期望为0.5左右

[1.99076511 1.44114769]