阿里历年双十一销售额数据拟合

import matplotlib.pyplot as plt
import numpy as np
from scipy.optimize import curve_fit


def f_fit(x, a, b, c):
    return a * x ** 2 + b * x + c


x = [1, 2, 3, 4, 5, 6, 7, 8, 9]
y = [9.36, 52, 191, 350, 571, 912, 1207, 1682, 2135]
p_fit, prov = curve_fit(f_fit, x, y)

print(p_fit)
print(f_fit(10, p_fit[0], p_fit[1], p_fit[2]))
plt.scatter(x, y)
x_line = np.arange(1, 9, 0.1)
y_line = p_fit[0] * x_line ** 2 + p_fit[1] * x_line + p_fit[2]
plt.plot(x_line, y_line)
plt.show()