グラフに線形トレンド線の代わりに3次関数を追加しようとしています

Question

xとyはデータリストで線形のトレンド線でうまくプロットしています。
また、立方体のトレンドラインを追加したいと思います。

import matplotlib.pyplot as plt 
x = (distanceList) 
y = (percentCopper) 
plt.scatter(x,y) 
title = "trendLine" 
xLabel = "Distance m" 
yLabel = "percent copper" 
plt.title (title, fontsize=10); 
plt.ylabel(yLabel, fontsize=10); 
plt.xlabel(xLabel, fontsize=10); 
fit = np.polyfit(x,y,1) 
fit_fn = np.poly1d(fit) 
plt.plot(x, y, '.', x, fit_fn(x), 'r') 
plt.xlim(0, 50) 
plt.ylim(0, 2.5) 
plt.show() 

Answer 1

だけnp.polyfit(x,y,3)を使用して、以下のコードのように、プロットに追加します。

import matplotlib.pyplot as plt 
import numpy as np 
x = np.array(range(50)) 
y = x**3/5000.0-x/5000.0 
plt.scatter(x,y) 
title = "trendLine" 
xLabel = "Distance m" 
yLabel = "percent copper" 
plt.title (title, fontsize=10); 
plt.ylabel(yLabel, fontsize=10); 
plt.xlabel(xLabel, fontsize=10); 
fit = np.polyfit(x,y,1) 
fit3 = np.polyfit(x,y,3) 
fit_fn = np.poly1d(fit) 
fit_fn3 = np.poly1d(fit3) 
plt.plot(x, y, '.', x, fit_fn(x), fit_fn3(x), 'r') 
plt.xlim(0, 50) 
plt.ylim(0, 2.5) 
plt.show()