You can use numpy's polyfit.
import numpy as np
from matplotlib import pyplot as plt
X = np.linspace(0, 100, 50)
Y = 23.24 + 2.2*X + 0.24*(X**2) + 10*np.random.randn(50) #added some noise
coefs = np.polyfit(X, Y, 2)
print(coefs)
p = np.poly1d(coefs)
plt.plot(X, Y, "bo", markersize= 2)
plt.plot(X, p(X), "r-") #p(X) evaluates the polynomial at X
plt.show()
Out:
[ 0.24052058 2.1426103 25.59437789]
与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…