Python：二次曲线拟合（节约生命法拟合）

有一组离散点，可以直接使用plot画出来

肉眼观察大致是个二次的曲线，但是假如想拟合出曲线参数，比如

y=ax^2+bx+c，想求出abc，就有点繁琐。人生苦短，python一行。

import matplotlib.pyplot as plt
import numpy as np

dj = [0,62,125,187,250,312,375,437,500,562,625]
dis = [1193,1233,1316,1384,1534,1680,1850,2069,2350,2680,3078]
plt.plot(dj,dis,color='red')
# plt.show()

coef = np.polyfit(dj, dis, 2)
y_fit = np.polyval(coef, dj)
plt.plot(dj, y_fit, 'g')
plt.show()

# 4.74112516e-03, -5.22165639e-02,  1.22032347e+03

绿线就是拟合的曲线，参数在coef里。