Python绘制3D电势球体报错IndexError：求排查与解决方案

错误根源分析 #

你的代码出现IndexError的核心原因是数据维度不匹配，同时存在球坐标计算、绘图逻辑的错误：

• 用随机点而非二维网格数据：plot_surface要求输入的X、Y、Z必须是二维网格数组，但你用np.random.uniform生成的是一维随机点，后续meshgrid(x,y)生成1000x1000的数组，而电势V还是1x1000的一维数组，维度无法对应。
• 球坐标公式错误：原代码中theta和phi的角色搞反，导致球体形状计算错误。
• 绘图参数传递错误：plot_surface的第三个参数应该是球体的Z坐标，而非电势V，电势值需要通过facecolors映射到表面颜色。

修正后的完整代码 #

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

def v_func(theta, r):
    E_0 = 1
    e_0 = 1
    a = 1
    Q_0 = 1
    return -E_0*r*np.cos(theta) + (2*E_0*a**3/r**2)*np.cos(theta) + Q_0/(4*np.pi*e_0*r)

fig, ax = plt.subplots(subplot_kw=dict(projection="3d"), constrained_layout=True)

r = 1
# 生成二维网格的theta（极角，0到pi）和phi（方位角，0到2pi）
theta = np.linspace(0, np.pi, 100)
phi = np.linspace(0, 2*np.pi, 100)
theta_grid, phi_grid = np.meshgrid(theta, phi)

# 正确的球坐标转笛卡尔坐标
x = r * np.sin(theta_grid) * np.cos(phi_grid)
y = r * np.sin(theta_grid) * np.sin(phi_grid)
z = r * np.cos(theta_grid)

# 计算每个网格点的电势值，得到二维数组
V = v_func(theta_grid, r)

# 归一化电势值用于颜色映射
norm = plt.Normalize(vmin=V.min(), vmax=V.max())
# 绘制球体表面，用V的归一化值映射颜色
surf = ax.plot_surface(x, y, z, facecolors=plt.cm.plasma(norm(V)), rstride=1, cstride=1)

# 添加颜色条
m = plt.cm.ScalarMappable(cmap=plt.cm.plasma, norm=norm)
m.set_array(V)
plt.colorbar(m, label="电势值")

ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
ax.set_title('球体表面电势分布')

plt.show()

关键修正点说明 #

• 改用网格数据：用np.linspace生成线性间隔的theta和phi数组，再通过meshgrid得到二维网格，确保X、Y、Z、V都是同维度的二维数组。
• 修正球坐标公式：调整theta和phi在x、y、z计算中的位置，符合标准球坐标定义（theta为极角，从z轴正方向向下测量）。
• 正确传递绘图参数：plot_surface的前三个参数是球体的x、y、z坐标，表面颜色由facecolors参数传入，通过归一化后的V值映射到plasma色卡。
• 修复颜色条关联：确保ScalarMappable正确关联V数组，让颜色条能准确对应电势值范围。

内容的提问来源于stack exchange，提问作者Ezzy Cross