用多项式模型进行数据拟合实验

最新推荐文章于 2024-06-12 21:00:00 发布

叶落，何语

最新推荐文章于 2024-06-12 21:00:00 发布

阅读量1.6k

点赞数

文章标签： python

本文链接：https://blog.csdn.net/weixin_44712846/article/details/105288742

版权

用多项式模型进行数据拟合实验

首先在文件lorenza.dat中读取出前20个数据，然后将这20个数据代入到公式中求出多项式系数，从而建立出多项式回归模型，最后通过建立出来的多项式回归模型画出P次多项式的数据拟合图。

# -*- coding: utf-8 -*-
"""
Created on Thu Mar 19 00:06:41 2020

@author: 侯明会
"""
import matplotlib.pyplot as plt
import numpy as np
N=20
data=[0]*N
#在文件中读取20个数据
fo=open("lorenza.dat","r+")
list_x=list(fo.read().splitlines(False))
fo.close()
for i in range(N):
    data[i]=float(list_x[i])

#多项式变换
def Polynomial(p):
    index=[]
    for i in range(N):
        index=index+[i]
    z1=np.polyfit(index,data,p)#返回P次多项式系数
    p1=np.poly1d(z1)#生成多项式
    #print(p1) #打印拟合的多项式
    yvals=p1(index) #拟合后的y值
    plot1=plt.plot(index,data,'r*',label='original values')
    plot2=plt.plot(index,yvals,label='polyfit values')
    #plt.axis([0,20,0,25])
    plt.xlabel('X ')
    plt.ylabel('Y')
    plt.legend(loc=3) #设置图示的位置
    plt.title('polyfitting') #设置标题
    plt.show() #显示图片
    return
temp=True #确保输入的P值不小于0
while temp:
    p=int(input("请输入P值："))
    if p<0:
        print("P值不能小于0，请重新输入")
    else:
        break
print(str(p)+"次多项式数据拟合图为：")
Polynomial(p)