Tkinter中Seaborn热力图无法显示的问题求助

核心问题分析 #

• 绘图上下文混淆：使用plt.ylim()全局pyplot函数而非当前figure的Axes对象设置坐标轴范围，导致Seaborn热力图未正确关联到Tkinter嵌入的figure。
• 变量名冲突：函数参数的x、y、z被循环内同名变量覆盖，虽不直接引发绘图问题，但降低代码可读性且存在潜在bug。
• 重复绘图未清理：每次点击按钮都会创建新Canvas，多次点击后绘图重叠，影响显示效果。

修改后的完整代码 #

import random
import numpy as np
import pandas as pd
import seaborn as sns
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from matplotlib.figure import Figure
from tkinter import *

root = Tk()
# 全局变量存储canvas，用于清理旧绘图
canvas = None

def HPS(x_entry, y_entry, z_entry): 
    global canvas
    NUMX = int(x_entry.get())
    NUMY = int(y_entry.get())
    NUMZ = int(z_entry.get())
    
    conc = np.zeros((NUMZ, NUMY, NUMX))

    # 重命名循环变量，避免覆盖参数
    for z_level in range(NUMZ):
        for y_row in range(NUMY):
            for x_col in range(NUMX):
                c = round(random.uniform(0.0, 1.0), 2)
                conc[z_level, y_row, x_col] = c
                 
    conc_sum = conc.sum(axis=0)
    df_sum = pd.DataFrame(conc_sum)

    # 创建figure和axes，使用对象方法操作
    figure = Figure(figsize=(4, 4))
    ax = figure.subplots()
    sns.heatmap(df_sum, cmap='RdBu', ax=ax)  # 指定ax参数，确保热力图绘制在正确的轴上
    ax.set_ylim(0, NUMY)  # 使用Axes对象的set_ylim替代plt.ylim
    
    # 清理旧的canvas
    if canvas is not None:
        canvas.get_tk_widget().destroy()
    
    # 嵌入新绘图到Tkinter
    canvas = FigureCanvasTkAgg(figure, master=root)
    canvas.draw()
    # 让绘图居中显示，使用sticky和columnspan配合grid布局
    canvas.get_tk_widget().grid(row=8, column=0, columnspan=3, sticky="nsew")
    # 配置列权重，让列随窗口拉伸
    root.grid_columnconfigure(0, weight=1)

# GUI部分
root.geometry("400x400")
frame1 = LabelFrame(root, text='Frame 1', padx=10, pady=10)
frame1.grid(row=0, column=0, padx=10, pady=10)

label1 = Label(frame1, text='NUMX = ') 
label1.grid(row=0, column=0)
label2 = Label(frame1, text='NUMY = ') 
label2.grid(row=1, column=0)
label3 = Label(frame1, text='NUMZ = ') 
label3.grid(row=2, column=0)

x_entry = Entry(frame1)    
y_entry = Entry(frame1)
z_entry = Entry(frame1)
x_entry.grid(row=0, column=2)
y_entry.grid(row=1, column=2)
z_entry.grid(row=2, column=2)

myButton = Button(root, text='Submit', command=lambda:HPS(x_entry, y_entry, z_entry))
myButton.grid(row=5, column=0, pady=10)

root.mainloop()

关键修改点说明 #

• 指定Seaborn绘图的Axes：调用sns.heatmap()时添加ax=ax参数，确保热力图绘制在我们创建的Axes对象上，而非默认的pyplot全局轴。
• 替换plt.ylim()为ax.set_ylim()：使用当前figure的Axes对象设置坐标轴范围，避免全局绘图上下文干扰。
• 修复变量名冲突：将循环内的z、y、x改为z_level、y_row、x_col，避免覆盖函数参数的输入控件对象。
• 清理旧绘图：每次点击按钮时先销毁之前的Canvas，防止多次点击导致绘图重叠。
• 优化布局居中：通过sticky="nsew"和grid_columnconfigure让绘图区域随窗口拉伸，保持居中显示。

内容的提问来源于stack exchange，提问作者Rathpr