Test Thailand coronavirus graph fitting

TH-covid19-curve_fit.py

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


x = [46,67,81,110,121,144,155,250,194,279,216,315,745,527]
j = 0
buckets = [0] * len(x)
m = 0
for i in x:
    m += i
    buckets[j] = m
    j += 1
buckets
y = list(range(1, len(x)+1))
# plt.plot( y,x)
# plt.plot( y,buckets)




# def exponential(x, a, b):
#     return a*np.exp(b*x)

x = np.array(y)
y = np.array(buckets)
# Function to calculate the exponential with constants a and b
def func(x, a, b):
    return a * np.exp(b * x)

params, param_cov = curve_fit(func, x, y)
print(params)
print(param_cov)

ans = (params[0]*(np.exp(params[1]*x)))
plt.plot(x, y, 'o', color ='red', label ="data")
plt.plot(x, ans, '--', color ='blue', label ="optimized data")
plt.legend()
plt.show()

https://colab.research.google.com/drive/1k4CfYsIwoAOAHnd_a15tFbm6AyxojiCx