python绘制热力图代码

下面是使用Python的matplotlib库绘制热力图的示例代码：

import numpy as np
import matplotlib.pyplot as plt

# 生成数据
data = np.random.rand(10, 10)

# 绘制热力图
fig, ax = plt.subplots()
im = ax.imshow(data, cmap='viridis')

# 添加颜色条
cbar = ax.figure.colorbar(im, ax=ax)

# 设置坐标轴标签和标题
ax.set_xticks(np.arange(data.shape[1]))
ax.set_yticks(np.arange(data.shape[0]))
ax.set_xticklabels(['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J'])
ax.set_yticklabels(['1', '2', '3', '4', '5', '6', '7', '8', '9', '10'])
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_title('Heatmap Example')

# 在每个单元格中添加数值标签
for i in range(data.shape[0]):
    for j in range(data.shape[1]):
        text = ax.text(j, i, '{:.2f}'.format(data[i, j]),
                       ha='center', va='center', color='w')

# 显示图像
plt.show()

该代码将生成一个随机数据的10x10热力图，其中每个单元格都显示一个随机数，同时在每个单元格中添加了数值标签。你可以根据实际需求修改代码和数据来绘制自己的热力图。