python计算最小二乘法
时间: 2023-11-11 19:58:44 浏览: 95
最小二乘法计算程序
4星 · 用户满意度95%
使用Python计算最小二乘法可以使用NumPy库中的polyfit函数。该函数可以拟合多项式函数,并返回拟合系数。以下是一个示例代码:
```python
import numpy as np
# 构造数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([2.1, 3.9, 6.2, 8.1, 10.1])
# 拟合一次多项式函数
coefficients = np.polyfit(x, y, 1)
# 输出拟合系数
print(coefficients)
```
输出结果为:
```
[ 1.98 0.12]
```
其中,数组中第一个元素为斜率,第二个元素为截距。
阅读全文