在 python 中绘制具有多个参数的数学函数的最佳方法？

Question

我是（理论）物理专业的学生，我对编程这方面的东西真的很陌生。我正在尝试在 Python 中绘制二维量子系统的径向薛定谔波函数，但我遇到了很多麻烦

import numpy as np
import scipy.special as ss
import matplotlib.pyplot as plt


#beta parameter definition
def beta(m, flux, gama):
    flux = np.linspace(0,1.0)
    return np.sqrt((m-flux)**2+gama**4)

#radial wave function definition
beta = beta(0.067*9.1E-31, 0.5, 1.5)
R=5.0E-9
r=np.linspace(0, 0.6)
ro = r/R
def radial (n,gama,beta, ro, R,r):
    return (1/R)*np.sqrt((ss.gamma(n+1)/2**beta*ss.gamma(n+beta+1)))*(gama*ro)**beta*np.exp(-(gama*ro)**2/4)*ss.eval_genlaguerre(n,beta,((gama*ro)**2/2))

sol = radial(0, 1.5, beta, ro, R,r)
plt.plot(ro, sol, 'b-')
plt.xlabel('r/R')
plt.ylabel('R(r)')
plt.title('Solução radial em fução da coordenada radial')
plt.legend("gamma=1,5")
plt.grid()
plt.show()

我想问：将多个参数传递给数学函数的最佳方法是什么？即使有多个参数和特殊功能，我也只能在绘图上得到一条直线。感谢您提供任何提示或帮助。

Answer 1

您正在将非布尔值输入到 genlaguerre 函数的布尔参数中 https://docs.scipy.org/doc/scipy/reference/generated/scipy.special.genlaguerre.html

def genlaguerre(n: int,
                alpha: float,
                monic: Optional[bool] = False) -> orthopoly1d)

您正在输入：

ss.genlaguerre(n=n,
               alpha=beta,
               monic=((gama*ro)**2/2)  #  <--- this is your issue
               )

Answer 2

函数ss.genlaguerre接受布尔值作为第三个参数，而不是数组。

https://docs.scipy.org/doc/scipy/reference/generated/scipy.special.genlaguerre.html#scipy.special.genlaguerre

在 python 中绘制具有多个参数的数学函数的最佳方法？

The best way to plot a math function with multiple parameters in python?

python

scientific-computing

python-3.x