Python 画热力图
时间: 2025-01-06 11:43:51 浏览: 61
### 如何使用 Python 绘制热力图
为了绘制热力图,可以利用 `matplotlib` 库和 `seaborn` 库的功能[^1]。
#### 安装必要的库
如果尚未安装这些库,则可以通过运行以下命令来完成安装:
```bash
pip install matplotlib seaborn
```
#### 导入所需库并准备数据
在开始绘图之前,先导入所需的库,并准备好要可视化的数据集。这里以随机生成的数据为例说明如何操作[^2]。
```python
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt
data = np.random.rand(10, 10) # 创建一个10x10的随机数矩阵作为示例数据
```
#### 设置热力图表参数
通过调整不同的参数选项来自定义热力图的表现形式。常用的几个重要参数包括但不限于:`data`, `cmap`, `annot`, `fmt`, `cbar`, `cbar_kws`, `xticklabels`, 和 `yticklabels` 等[^3]。
- `data`: 输入的二维数组或 DataFrame 类型的数据源;
- `cmap`: 设定颜色映射方案,默认情况下会采用渐变色彩模式;
- `annot`: 如果设为 True ,则会在每个方格内标注具体的数值大小;
- `fmt`: 控制数值显示格式字符串;
- `cbar`: 显示与否右侧的颜色比例尺;
- `cbar_kws`: 对颜色条附加更多样式控制字典;
- `xticklabels/yticklabels`: 可以为 X 轴/Y 轴指定自定义标签列表;
下面是一段完整的代码片段展示如何应用上述提到的各种配置项来构建一张美观实用的热力图:
```python
sns.set_theme(style="whitegrid") # 设置主题风格可选
plt.figure(figsize=(8, 6))
ax = sns.heatmap(
data,
cmap='viridis',
annot=True,
fmt=".2f",
cbar=True,
cbar_kws={'label': 'Value'},
)
plt.title('Heatmap Example Using Seaborn')
plt.xlabel('X Axis Label')
plt.ylabel('Y Axis Label')
plt.tight_layout()
plt.show()
```
这段代码不仅展示了基本功能的应用方式,还加入了额外的主题设定(`set_theme`)以及图形尺寸调整(`figure`)等内容,使得最终呈现出来的图像更加清晰易读[^4]。
阅读全文
相关推荐


















