带数值标签的热力图Python代码
时间: 2023-05-29 14:03:25 浏览: 209
Python绘制热力图示例
4星 · 用户满意度95%
以下是一个简单的Python代码示例,生成带数值标签的热力图:
```python
import matplotlib.pyplot as plt
import numpy as np
# 生成随机矩阵
data = np.random.rand(5, 5)
# 绘制热力图
fig, ax = plt.subplots()
im = ax.imshow(data)
# 添加数值标签
for i in range(len(data)):
for j in range(len(data[0])):
text = ax.text(j, i, round(data[i][j], 2), ha="center", va="center", color="w")
# 设置图形属性
ax.set_xticks(np.arange(len(data[0])))
ax.set_yticks(np.arange(len(data)))
ax.set_xticklabels(['A', 'B', 'C', 'D', 'E'])
ax.set_yticklabels(['1', '2', '3', '4', '5'])
ax.set_title("Heatmap with Labels")
# 显示图形
plt.show()
```
这个示例中,我们使用了NumPy生成了一个5x5的随机矩阵,并使用Matplotlib的imshow方法绘制了一个简单热力图。然后,我们使用两个嵌套的for循环,将每个数据点的数值标签添加到热力图上。最后,我们设置了轴标签和图形标题,并将图形显示出来。
阅读全文