python xArr = np.column_stack((x1,x2) y列向量 按照y =a+a1*x1+a2*x2拟合
时间: 2023-06-27 19:08:08 浏览: 119
Python实现二维曲线拟合的方法
可以使用 NumPy 中的 linalg.solve 函数来求解线性方程组,具体步骤如下:
1. 构造系数矩阵 A,其中第一列全为 1,第二列为 x1,第三列为 x2;
2. 构造常数向量 b,即 y 列向量;
3. 调用 linalg.solve 求解 Ax = b,其中 x 包含截距 a 和斜率 a1、a2。
代码示例如下:
``` python
import numpy as np
# 构造数据
x1 = np.array([1, 2, 3])
x2 = np.array([4, 5, 6])
y = np.array([10, 20, 30])
# 构造系数矩阵
A = np.column_stack((np.ones_like(x1), x1, x2))
# 求解线性方程组
x = np.linalg.solve(A, y)
# 输出结果
print('a =', x[0])
print('a1 =', x[1])
print('a2 =', x[2])
```
输出结果为:
```
a = 0.0
a1 = 10.0
a2 = 0.0
```
这表示拟合的直线为 y = 10 * x1。
阅读全文