python 画拟合曲线

Python是一种通用高级编程语言，具有简单易用、效力高、可移植等优点，广泛利用于科学计算、机器学习、Web开发等领域。其中绘图是Python的重要利用之一，下面我们来看一下怎样使用Python画拟合曲线。

# 导入相关库
import numpy as np
import matplotlib.pyplot as plt
# 生成数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([0.5, 2.5, 2, 4.5, 3.5])
# 一次函数拟合
z1 = np.polyfit(x, y, 1)
p1 = np.poly1d(z1)
# 二次函数拟合
z2 = np.polyfit(x, y, 2)
p2 = np.poly1d(z2)
# 绘制原始散点图和拟合曲线
plt.scatter(x, y, c='r', label='data')
plt.plot(x, p1(x), '--', linewidth=2, label='linear')
plt.plot(x, p2(x), '-.', linewidth=2, label='quadratic')
plt.legend()
plt.show()

在这段代码中，我们使用numpy库生成了一些数据，分别代表x轴和y轴的值。然后使用numpy的polyfit方法用一次函数和二次函数分别进行拟合，得到拟合曲线p1和p2。最后使用matplotlib.pyplot库中的scatter和plot方法将原始散点图和拟合曲线绘制出来。

通过这类方式，我们可以很方便的绘制出数据的拟合曲线，来更加直观的展现数据特点和趋势，有助于我们做出更好的决策。