실제 데이터와 같은 플롯에서 다항식 회귀를 플로팅

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 

아무도 나를 원래의 데이터와 동일한 플롯에 회귀 polynomail이 5 차의 결과를 플롯 도와 드릴까요 단순히 다항식 회귀 수행한다. tcaswell에서 제공하는 감사 GTPE

편집

코드는 완벽하게 작동합니다. 내 유일한 질문은 다항식이 왜 그렇게 각성되어 나오는가하는 것입니다.

이

Answer 1

그냥 다시 plot 전화 : 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() 

여러 가지 인자를 사용하는 다른 방법이 np.polyfit 문서에서 볼 수 있습니다 여기 내 예제 이미지입니다.