原文链接: 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]