我正在尝试为我定义的函数中的参数设置默认值。我还希望另一个参数具有依赖于另一个参数的默认值。在我的示例中,我试图绘制氢的量子力学波函数,但您不需要了解物理学即可为我提供帮助。
def plot_psi(n,l,start=(0.001*bohr),stop=(20*bohr),step=(0.005*bohr)):
这里n
的主要量子数l
是角动量,start,stop,step
将是我计算出的数组。但是我需要的是的默认值stop
实际上取决于n
,因为n
它将影响wavefunction的大小。
def plot_psi(n,l,start=(0.001*bohr),stop=((30*n-10)*bohr),step=(0.005*bohr)):
将会是我想要的,但是n
由于该行还不完整,所以尚未定义。有什么办法吗?还是有其他安排的想法?谢谢
使用None
作为默认值,并计算值的函数里,像这样的
def plot_psi(n, l, start=(0.001*bohr),stop=None,step=(0.005*bohr)):
if stop is None:
stop = ((30*n-10)*bohr)
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句