csv - 在与真实数据相同的图中绘制多项式回归

Question

我有一些代码片段，可以读取两个 csv 并使用 matplotlib.pyplot 绘制它们，并对相同的两个 csv 执行多项式回归。我想要做的是将数据和多项式回归绘制在同一张图上。

import matplotlib.pyplot as plt
import csv
import numpy as np

datax=np.genfromtxt('Delta R.csv')
datay=np.genfromtxt('Example R.csv')


plt.title ('Test graph ')
plt.xlabel('x axis')
plt.ylabel('y axis ')

plt.plot(datax, datay,'o-')

plt.show()

我的第二段代码执行简单的多项式回归

import matplotlib.pyplot as plt
import csv
import numpy as np

datax=np.genfromtxt('Delta R.csv')
datay=np.genfromtxt('Example R.csv')

z = np.polyfit(datax,datay,5) 

print z

任何人都可以帮我在与原始数据相同的图中绘制五阶多项式回归的结果。感谢GTPE

编辑

tcaswell 提供的代码运行良好。我唯一的问题是为什么多项式在某些地方表现得如此有角度。这是我的示例图像:

Answer 1

只需调用 plot再次:

datax=np.genfromtxt('Delta R.csv')
datay=np.genfromtxt('Example R.csv')

z = np.polyfit(datax,datay,5) 
p = np.poly1d(z)

plt.title ('Test graph ')
plt.xlabel('x axis')
plt.ylabel('y axis ')

plt.plot(datax, datay,'o-')
plt.plot(datax, p(datax), '-')

datax_os = np.linspace(np.min(datax), np.max(datax), 1024)
plt.plot(datax_os, f(datax_os)) # 'smoother' line

plt.show()

使用多个参数的替代方法 plot可以在 np.polyfit 中看到文档。