如何使用Pandas日历热图绘制特定日期

pip install pandas

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.dates as mdates

# 创建日期范围
dates = pd.date_range(start='2022-01-01', end='2022-12-31')

# 创建随机数据
data = np.random.randint(low=0, high=100, size=len(dates))

# 创建DataFrame对象
df = pd.DataFrame({'Date': dates, 'Data': data})

df.set_index('Date', inplace=True)

# 将数据按照月份和日期进行重塑
df_heatmap = df['Data'].groupby([df.index.month, df.index.day]).mean().unstack()

# 创建日历热图
fig, ax = plt.subplots(figsize=(12, 6))
ax.imshow(df_heatmap, cmap='YlOrRd')

# 设置x轴和y轴刻度
ax.set_xticks(np.arange(0, 12, 1))
ax.set_yticks(np.arange(0, 31, 1))

# 设置x轴和y轴标签
ax.set_xticklabels(range(1, 13))
ax.set_yticklabels(range(1, 32))

# 设置x轴和y轴刻度位置
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')

# 设置日期格式
ax.xaxis.set_major_formatter(mdates.DateFormatter('%b'))

# 添加颜色条
cbar = plt.colorbar()
cbar.set_label('Data')

# 添加标题和标签
plt.title('Calendar Heatmap')
plt.xlabel('Month')
plt.ylabel('Day')

# 显示图形
plt.show()