使用matplotlib和PolyCollection - Python的时间线栏

pip install matplotlib

import matplotlib.pyplot as plt
from matplotlib.collections import PolyCollection
import numpy as np

# 创建时间序列数据
dates = np.array(['2022-01-01', '2022-02-01', '2022-03-01', '2022-04-01', '2022-05-01'])
values = np.array([10, 20, 15, 25, 30])

# 将日期转换为数字
num_dates = np.arange(len(dates))

# 创建多边形集合
verts = list(zip(num_dates, values))
poly = PolyCollection([verts], facecolors='blue', alpha=0.6)

# 创建图表
fig, ax = plt.subplots()
ax.add_collection(poly)

# 设置x轴刻度和标签
ax.set_xticks(num_dates)
ax.set_xticklabels(dates, rotation=45)

# 设置y轴范围
ax.set_ylim(0, max(values) + 5)

# 添加标题和标签
ax.set_title('Timeline')
ax.set_xlabel('Date')
ax.set_ylabel('Value')

# 显示图表
plt.show()