python二元线性拟合
时间: 2023-10-02 20:12:05 浏览: 165
python如何实现数据的线性拟合
二元线性拟合是指通过拟合一条直线来描述两个变量之间的线性关系。在Python中,可以使用最小二乘法来进行二元线性拟合。下面是一个示例代码:
import numpy as np
from numpy.linalg import lstsq
# 假设我们有以下数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([2, 4, 6, 8, 10])
# 添加偏置项
X = np.vstack([x, np.ones(len(x))]).T
# 使用最小二乘法进行拟合
m, c = lstsq(X, y, rcond=None)
# 输出拟合结果
print("斜率m:", m)
print("截距c:", c)
这段代码中,我们首先将自变量x和常数项1堆叠起来形成矩阵X,然后使用最小二乘法求解得到斜率m和截距c,最后打印输出结果。
阅读全文