用python中matplotolb求最小二乘法

可以使用python中的numpy和matplotlib库来实现最小二乘法。

首先，导入必要的库：

import numpy as np
import matplotlib.pyplot as plt

然后，定义x和y的值：

x = np.array([1, 2, 3, 4, 5])
y = np.array([2, 4, 5, 4, 5])

接下来，可以使用numpy中的polyfit函数来进行最小二乘法的计算：

p = np.polyfit(x, y, 1)  # 1代表一次函数

这里的p是一个包含最小二乘法系数的数组，第一个元素是斜率，第二个元素是截距。

最后，可以使用matplotlib来绘制出拟合的直线：

plt.scatter(x, y)
plt.plot(x, p[0]*x + p[1], color='red')
plt.show()

其中，scatter函数用于绘制散点图，plot函数用于绘制拟合的直线，color参数用于设置直线的颜色。

完整代码如下：

import numpy as np
import matplotlib.pyplot as plt

x = np.array([1, 2, 3, 4, 5])
y = np.array([2, 4, 5, 4, 5])

p = np.polyfit(x, y, 1)

plt.scatter(x, y)
plt.plot(x, p[0]*x + p[1], color='red')
plt.show()