Sympy lambdify处理常数函数时维度不匹配引发绘图报错问题

这不是SymPy的缺陷，是sympy.lambdify的默认设计行为：当待转换表达式完全不含指定自变量x（即纯常数表达式）时，生成的NumPy目标函数不会对输入的自变量数组做广播处理，直接返回常量对应的标量值，最终导致绘图时传入的x、y数组维度不匹配，抛出ValueError。

以下三种方案均可解决问题，按需选择即可：

• 最小改动方案：获取y_range后自动做维度对齐，不需要修改原有lambdify逻辑，兼容所有SymPy版本、所有合法输入表达式
  仅需新增一行广播逻辑，利用NumPy的自动广播特性，无论返回值是标量还是同维度数组都能正常处理：

  import sympy as sy
  import numpy as np
  import matplotlib.pyplot as plt
  
  x = sy.symbols('x')
  func = 5
  x_range = np.linspace(0, 10, 100)
  F = sy.lambdify(x, func, "numpy")
  y_range = F(x_range)
  # 自动对齐y与x的维度
  y_range = y_range * np.ones_like(x_range)
  plt.plot(x_range, y_range)
  plt.show()
  

• 前置判断方案：表达式转换前先判断是否为纯常数，走分支处理
  利用SymPy自带的has方法检测表达式是否包含自变量，纯常数直接生成对应长度的固定值数组，逻辑更直白：

  import sympy as sy
  import numpy as np
  import matplotlib.pyplot as plt
  
  x = sy.symbols('x')
  func = 5
  x_range = np.linspace(0, 10, 100)
  if func.has(x):
      F = sy.lambdify(x, func, "numpy")
      y_range = F(x_range)
  else:
      # 纯常数直接生成同长度固定值数组
      y_range = np.full_like(x_range, float(func))
  plt.plot(x_range, y_range)
  plt.show()
  

• 参数适配方案：SymPy 1.12及以上版本可以在调用lambdify时传入cse=True参数，该模式下生成的函数会自动处理常量的广播逻辑，不需要额外写对齐代码，但低版本SymPy不支持该参数的对应特性，使用前需要确认环境版本。

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