plt python 自己制定cmap_用 Python+SciPy 可视化定态氢原子波函数（二）

之前绘制了

截面的定态氢原子波函数的密度图，链接如下：

王欢：用 Python+SciPy 可视化定态氢原子波函数​zhuanlan.zhihu.com

这次，尝试着绘制定态氢原子波函数的轨道，其实主要将是三维的球谐函数部分绘制出来。在上次的介绍中我们发现，可以直接调用 scipy 库中的 special.sph_harm() 函数，但是参数的顺序和意义需要仔细。有了上次的经验，应该可以少走一些弯路。闲话少说，上代码：

import numpy as np
from scipy.special import sph_harm 
import matplotlib.pyplot as plt
from matplotlib import cm 
from mpl_toolkits.mplot3d import Axes3D


l = float(input("Enter l value: Note that l should >= 0 n"))
m = float(input("Enter m value: Note that m should in [-l, l] n"))

'''
set angles (theta and phi). 
Note: in Scipy, theta and phi are azimuthal angle and polar angle, respectively.
'''
theta = np.linspace(0, 2 * np.pi, 181)