seaborn绘制热力图 添加自定义颜色条,颜色为绿色、黄色、橙色、红色,颜色条的刻度间隔为50 单元格显示整数
时间: 2024-09-30 12:11:09 浏览: 78
python seaborn matplotlib 绘制热力图方法
在Seaborn库中创建热力图并添加自定义颜色条,你可以按照以下步骤操作:
1. 首先,确保安装了`seaborn`库,如果尚未安装,可以使用`pip install seaborn`命令。
2. 导入所需的库:
```python
import seaborn as sns
import numpy as np
import matplotlib.pyplot as plt
```
3. 创建数据矩阵,这里我们使用numpy生成一些随机数据:
```python
data = np.random.randint(0, 150, size=(10, 10))
```
4. 使用`heatmap`函数绘制热力图,并设置颜色条的颜色范围和刻度间隔:
```python
# 设置颜色映射,将绿色、黄色、橙色、红色对应到特定数值区间
cmap = sns.diverging_palette(50, 250, n=4, center='light')
# 创建热力图
sns.heatmap(data, cmap=cmap, annot=True, fmt="d") # 'fmt="d"'用于显示整数
# 自定义颜色条
plt.colorbar(label='Value', ticks=np.arange(0, 151, 50)) # 刻度间隔为50
```
5. 最后,显示图形:
```python
plt.show()
```
这将展示一个热力图,其中颜色条从绿色渐变到红色,每个50单位递增,单元格显示的是整数。
阅读全文