输出一个带有数值的热力图
时间: 2024-03-03 21:47:43 浏览: 52
好的,以下是一个 Python 中用 seaborn 库绘制带有数值的热力图的示例代码:
```python
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt
# 定义数据
data = np.random.rand(5, 5)
# 绘制热力图
ax = sns.heatmap(data, cmap='YlGnBu', annot=True, fmt='.2f')
# 设置坐标轴标签
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
# 显示图形
plt.show()
```
这段代码中,我们首先生成了一个 5x5 的随机数据矩阵,然后使用 `heatmap` 函数将其绘制成热力图。`cmap` 参数指定了使用的颜色映射(colormap),这里我们使用了 `'YlGnBu'` colormap,可以根据需要选择其他的 colormap。`annot` 参数指定了是否在热力图上显示数值标签,这里我们将其设置为 `True`。`fmt` 参数指定了数值标签的格式,这里我们将其格式化为保留小数点后两位的浮点数(`.2f`)。接着,我们使用 `set_xlabel` 和 `set_ylabel` 方法设置坐标轴标签。最后,我们使用 `show` 函数显示图形。
阅读全文