使用matplotlib (或seaborn)绘制d轨道图

import numpy as np
import matplotlib.pyplot as plt

fig, ax = plt.subplots()

def d_orbital(r, theta, phi):
    # d轨道的波函数表达式
    # 根据具体的波函数表达式进行计算
    return wave_function

r = np.linspace(0, 1, 100)  # r轴的取值范围
theta = np.linspace(0, np.pi, 100)  # θ轴的取值范围
phi = np.linspace(0, 2*np.pi, 100)  # φ轴的取值范围

R, Theta, Phi = np.meshgrid(r, theta, phi)  # 生成网格数据
X = R * np.sin(Theta) * np.cos(Phi)  # x坐标
Y = R * np.sin(Theta) * np.sin(Phi)  # y坐标
Z = R * np.cos(Theta)  # z坐标

wave_function = d_orbital(R, Theta, Phi)  # 计算波函数值

ax.plot_surface(X, Y, Z, cmap='viridis', edgecolor='none', alpha=0.8)

ax.set_title('d轨道图')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')

plt.show()