如何在Python中复现Matlab风格的球面壳密度曲面图

我完全懂你想要的效果——在Matlab里用surf轻松做出了带密度颜色映射的球面壳图，换到Matplotlib的plot_surface就卡壳了，还好你已经用scatter摸到了门道，其实只要调整几个参数，plot_surface就能完美复现Matlab的效果，甚至更灵活。

咱们先直接上能跑通的代码，再慢慢拆解细节：

import matplotlib.pyplot as plt
import numpy as np

# 补全你代码里缺失的a、b值（示例用1和2，可替换为你的实际值）
a = 1
b = 2

# 你的原始球面坐标设置
N_r = 21; N_theta = 18; N_phi= 36;
r_index = N_r-1;

[phi,theta,r_sphere] = np.meshgrid(
    np.linspace(0,2*np.pi,N_phi),
    np.linspace(0,np.pi,N_theta),
    np.linspace(a,b,N_r)
);

# 球面壳的X/Y/Z坐标
X = r_sphere[:,:,r_index] * np.sin(theta[:,:,r_index]) * np.cos(phi[:,:,r_index]);
Y = r_sphere[:,:,r_index] * np.sin(theta[:,:,r_index]) * np.sin(phi[:,:,r_index]);
Z = r_sphere[:,:,r_index] * np.cos(theta[:,:,r_index]);

# 密度变量
rho = 1/r_sphere**2 * np.sin(theta) * np.cos(theta) * np.sin(phi)

# 核心：复现Matlab surf效果的plot_surface调用
fig1 = plt.figure(figsize=(16,9),dpi=80)
ax = fig1.add_subplot(projection="3d")

# 关键参数说明：
# - `array`：传入你的密度切片rho[:,:,r_index]，对应Matlab surf的第四参数
# - `cmap`：指定颜色映射，这里用你scatter里的'bwr'，和Matlab默认风格匹配
# - `edgealpha`：对应Matlab的EdgeAlpha，设置边的透明度
surf = ax.plot_surface(X, Y, Z, array=rho[:,:,r_index], cmap='bwr', edgealpha=0.2)

# 添加颜色条，和你scatter的设置一致
fig1.colorbar(surf, ax=ax, shrink=0.5, aspect=5)

# 强制坐标轴等比例，避免球面被拉伸变形（Matlab默认会处理，Matplotlib需要手动开）
ax.set_aspect('equal')

plt.show()

为什么你之前的plot_surface没效果？ #

Matlab的surf可以直接把颜色数据当第四位置参数，但Matplotlib的plot_surface需要明确用array参数传入颜色映射数据，同时得指定cmap才能让颜色生效。你之前的代码没传这俩参数，所以曲面只会用默认颜色，看不到密度对应的色彩变化。

额外优化建议 #

1. 提升曲面平滑度：如果觉得曲面不够细腻，可以增大N_theta和N_phi的数值（比如改成36和72），这样生成的面片更多，球面会更光滑。
2. 自定义颜色范围：如果想固定颜色的上下限（比如避免异常值影响颜色映射），可以用Normalize来手动设置：

   from matplotlib.colors import Normalize
   # 自己指定vmin和vmax，比如用rho全局的极值
   norm = Normalize(vmin=np.min(rho), vmax=np.max(rho))
   surf = ax.plot_surface(X, Y, Z, array=rho[:,:,r_index], cmap='bwr', edgealpha=0.2, norm=norm)
   

3. 匹配Matlab的其他细节：如果想让坐标轴标签、视角和Matlab一致，可以用ax.set_xlabel('X')等设置标签，用ax.view_init(elev=30, azim=45)调整视角。

这样跑出来的图就和你在Matlab里做的几乎一模一样了——是带连续颜色映射的曲面，而不是scatter的点云效果。

备注：内容来源于stack exchange，提问作者Researcher R