向matplotlib颜色栏添加自定义刻度线

import matplotlib.pyplot as plt
import matplotlib.colors as mcolors
from matplotlib.ticker import MultipleLocator, FormatStrFormatter

# 生成示例数据
data = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

# 绘制热力图
plt.imshow(data, cmap='hot')

# 创建颜色栏
cbar = plt.colorbar()

# 创建自定义刻度线
locator = MultipleLocator(1)  # 设置刻度线间隔为1
formatter = FormatStrFormatter('%d')  # 设置刻度线标签格式为整数
cbar.locator = locator
cbar.formatter = formatter

# 更新颜色栏
cbar.update_ticks()

# 显示图形
plt.show()