基于PySide6/SymPy/Matplotlib的3D函数绘图器维度错误求助

问题根源 #

plot_surface要求输入的Z必须是二维数组（对应X/Y的网格点），但你用SymPy解析后生成的Z大概率是一维数组，或者X/Y本身是一维数据未转换为网格格式，导致维度不匹配触发报错。

解决步骤 #

1. 生成网格数据：用numpy.meshgrid将一维的X、Y数组转换为二维网格矩阵，这是plot_surface的核心要求。
2. 安全解析函数：用SymPy的parse_expr解析用户输入的函数字符串，再通过lambdify转换为支持numpy数组运算的数值函数，避免手动eval的安全风险。
3. 嵌入GUI绘图：使用Matplotlib的FigureCanvasQTAgg将绘图画布嵌入PySide6窗口，替代SymPy自带的独立plot3d窗口。

修正后的简化代码示例 #

import sys
import numpy as np
from sympy import symbols, parse_expr, lambdify
from PySide6.QtWidgets import (QApplication, QMainWindow, 
                               QVBoxLayout, QWidget)
from matplotlib.backends.backend_qtagg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure

class FuncPlotter(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("3D函数绘图器")
        central_widget = QWidget()
        self.setCentralWidget(central_widget)
        self.layout = QVBoxLayout(central_widget)
        
        # 初始化Matplotlib画布
        self.figure = Figure(figsize=(8, 6))
        self.canvas = FigureCanvas(self.figure)
        self.layout.addWidget(self.canvas)
        
        # 绘制示例3D函数
        self.render_3d_plot("x**2 + y**2")

    def render_3d_plot(self, func_str):
        # 定义符号变量
        x, y = symbols('x y')
        # 解析用户输入的函数
        sym_func = parse_expr(func_str)
        # 转换为numpy兼容的数值函数
        num_func = lambdify((x, y), sym_func, 'numpy')
        
        # 生成一维坐标点
        x_range = np.linspace(-5, 5, 100)
        y_range = np.linspace(-5, 5, 100)
        # 转换为二维网格
        X, Y = np.meshgrid(x_range, y_range)
        # 计算二维Z数组
        Z = num_func(X, Y)
        
        # 绘制3D曲面
        self.figure.clear()
        ax = self.figure.add_subplot(111, projection='3d')
        ax.plot_surface(X, Y, Z, cmap='coolwarm')
        ax.set_xlabel('X')
        ax.set_ylabel('Y')
        ax.set_zlabel('Z')
        
        self.canvas.draw()

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = FuncPlotter()
    window.show()
    sys.exit(app.exec())

关键注意点 #

• 绝对不要直接用eval解析用户输入的函数字符串，parse_expr+lambdify是更安全的方案，能避免恶意代码执行。
• 必须确保Z的维度与X、Y完全匹配（都是N×M的二维数组），否则plot_surface会抛出维度错误。
• 每次重新绘图时调用self.figure.clear()，避免画布上残留旧的绘图内容。

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