在matplotlib.pyplot中绘制带日期的线性回归图

import matplotlib.pyplot as plt
import matplotlib.dates as mdates
import numpy as np
from datetime import datetime

dates = ['2022-01-01', '2022-01-02', '2022-01-03', ...]  # 日期数据
values = [10, 15, 12, ...]  # 对应的数值数据

dates = [datetime.strptime(date, '%Y-%m-%d').date() for date in dates]

model = np.polyfit(mdates.date2num(dates), values, 1)
line = np.poly1d(model)

fig, ax = plt.subplots()
ax.scatter(dates, values, color='blue', label='Data')
ax.plot(dates, line(mdates.date2num(dates)), color='red', label='Linear Regression')

ax.xaxis.set_major_locator(mdates.MonthLocator())  # 设置主要刻度为月份
ax.xaxis.set_major_formatter(mdates.DateFormatter('%Y-%m'))  # 设置刻度标签格式为年-月
plt.xticks(rotation=45)  # 旋转刻度标签，使其不重叠

ax.legend()
plt.title('Linear Regression with Dates')

plt.show()