seaborn绘制热力图,调整热力图格子的大小
时间: 2023-08-20 21:07:13 浏览: 1501
要调整热力图格子的大小,可以使用seaborn中的heatmap函数,其中的参数square可以控制热力图的格子是否为正方形,如果为True,则格子为正方形,否则为长方形。默认情况下,该参数为True,即格子为正方形。
如果希望调整格子的大小,可以设置该参数为False,并调整热力图的宽度和高度,以及热力图中每个格子的宽度和高度。示例代码如下:
```python
import seaborn as sns
import matplotlib.pyplot as plt
# 生成数据
data = sns.load_dataset("flights").pivot("month", "year", "passengers")
# 绘制热力图
fig, ax = plt.subplots(figsize=(8, 6))
sns.heatmap(data, cmap="YlGnBu", annot=True, fmt="d",
linewidths=.5, ax=ax, square=False,
cbar_kws={"shrink": .5},
yticklabels=True, xticklabels=True)
# 调整热力图的宽度和高度
plt.subplots_adjust(left=0.15, bottom=0.25)
# 调整每个格子的宽度和高度
ax.set_xticklabels(ax.get_xticklabels(), rotation=45, ha="right")
bottom, top = ax.get_ylim()
ax.set_ylim(bottom + 0.5, top - 0.5)
```
在上述代码中,我们通过设置square参数为False来调整格子的大小,然后通过subplots_adjust方法调整热力图的宽度和高度,最后通过set_ylim方法调整每个格子的宽度和高度。运行该代码可以得到一个调整了格子大小的热力图。
阅读全文