バレンタインチョコの個数を線形回帰で予測してみました。

import numpy as np
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression

chocolates = np.array([0, 1]).reshape(1, -1).T
year = np.array([2021, 2022]).reshape(1, -1).T
model = LinearRegression()
model.fit(year, chocolates)
future = np.linspace(2021, 2050, 30).reshape(1, -1).T
plt.scatter(year, chocolates, label="Observed")
plt.plot(future, model.predict(future), label="Predicted")
plt.title("Linear Regression")
plt.xlabel("Year")
plt.ylabel("Chocolates")
plt.grid()
plt.legend()
plt.show()