Tkinter with matplotlib - Heatmap colorbar不能与其他轴一起清除

import tkinter as tk
from matplotlib.figure import Figure
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg

# 创建Tkinter窗口
root = tk.Tk()

# 创建matplotlib的figure对象
fig = Figure()

# 在figure对象上创建子图
ax = fig.add_subplot(111)

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

# 绘制其他轴
# ...

# 创建FigureCanvasTkAgg对象，将figure对象绑定到Tkinter窗口
canvas = FigureCanvasTkAgg(fig, master=root)
canvas.draw()
canvas.get_tk_widget().pack()

# 清除图形并重新绘制
def clear_heatmap():
    ax.clear()
    # 绘制其他轴
    # ...
    heatmap = ax.imshow(new_data, cmap='hot')
    canvas.draw()

# 创建清除按钮
clear_button = tk.Button(root, text="Clear Heatmap", command=clear_heatmap)
clear_button.pack()

# 运行Tkinter主循环
root.mainloop()

import tkinter as tk
import matplotlib.pyplot as plt

# 创建Tkinter窗口
root = tk.Tk()

# 创建figure对象和子图
fig, ax = plt.subplots()

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

# 绘制其他轴
# ...

# 创建颜色条
colorbar = plt.colorbar(heatmap)

# 清除颜色条并重新绘制
def clear_colorbar():
    colorbar.remove()
    # 绘制其他轴
    # ...
    heatmap = ax.imshow(new_data, cmap='hot')
    colorbar = plt.colorbar(heatmap)

# 创建清除按钮
clear_button = tk.Button(root, text="Clear Colorbar", command=clear_colorbar)
clear_button.pack()

# 将matplotlib图形嵌入到Tkinter窗口
canvas = FigureCanvasTkAgg(fig, master=root)
canvas.draw()
canvas.get_tk_widget().pack()

# 运行Tkinter主循环
root.mainloop()