python的多项式画线
时间: 2023-11-03 11:00:54 浏览: 173
python 画线
要在Python中画多项式曲线,可以使用numpy库中的polyfit函数进行多项式拟合,并使用matplotlib库进行画图。
首先,使用polyfit函数拟合多项式并获得多项式系数an。然后,使用poly1d函数将系数an传入,生成一个多项式对象p1。
接下来,可以使用polyval函数传入多项式系数an和源数据点的横坐标x,得到通过多项式计算后的源数据点的纵坐标。最后,使用matplotlib库来绘制曲线。
示例代码如下所示:
import numpy as np
import matplotlib.pyplot as plt
# 源数据点
x = np.arange(-1.5, 1.6, 0.5)
y = [-4.45, -0.45, 0.55, 0.05, -0.44, 0.54, 4.55]
# 多项式拟合
an = np.polyfit(x, y, 3)
# 生成多项式对象
p1 = np.poly1d(an)
# 计算拟合后的纵坐标
y_fit = np.polyval(an, x)
# 绘制曲线
plt.plot(x, y, 'o', label='原始数据')
plt.plot(x, y_fit, label='拟合曲线')
plt.legend()
plt.show()
阅读全文