sns.heatmap(data, ax=ax)设置colorbar刻度

调整 Seaborn 热力图中 Colorbar 的刻度
为了调整 seaborn 中 sns.heatmap 函数绘制的热力图里 colorbar 的刻度，可以利用 Matplotlib 提供的功能来进一步定制化图表。具体来说，在创建完热力图之后获取其对应的 figure 对象并操作其中的颜色条。
下面是一个具体的例子展示如何设置 colorbar 刻度：
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
from numpy import arange

data = pd.DataFrame({
'A': [0.1, 0.2],
'B': [0.3, 0.4]
})

h = sns.heatmap(
data=data,
cmap='YlGnBu',
cbar=False, # 关闭默认颜色条以便后续自定义
annot=True,
vmin=0,
vmax=1
)

# 创建新的颜色条，并指定位置和尺寸
cb = h.figure.colorbar(h.collections[0], ticks=arange(0, 1.1, 0.1))

# 设置colorbar刻度字体大小
cb.ax.tick_params(labelsize=10)

# 设置colorbar标签名称
cb.set_label('Color Intensity')

plt.show()

在这个实例中，通过传递给 figure.colorbar() 方法一个带有特定 tick 参数的对象来自定义 colorbar 上显示的具体数值范围[^1]。这里使用了 NumPy 库中的 arange 函数生成一系列从 0 至 1 步长为 0.1 的浮点数作为刻度值。