Python plot_surface设置Y轴以0为中心的柱坐标绘图问题

听起来你是想把Y轴（对应rho/P）的零点放在中间，左侧显示rho从10到0的反向刻度，右侧显示rho从0到10的正向刻度——这样就能同时容纳Z正负对应的磁场数据，对吧？我之前也遇到过类似的需求，给你几个可行的方案：

方案1：重新映射Y轴坐标（推荐） #

因为rho本身是正的，而Z有正负，我们可以把Y轴的正负对应Z的正负，Y的绝对值对应rho的数值，这样Y轴自然就会以0为中心，左边是rho从10到0，右边是0到10。具体步骤如下：

1. 先生成原始的rho和Z的meshgrid：

import numpy as np
import matplotlib.pyplot as plt

# 生成rho（0到10）和Z（-10到10）的网格
rho = np.linspace(0, 10, 50)
Z = np.linspace(-10, 10, 50)
Z_mesh, rho_mesh = np.meshgrid(Z, rho)

# 假设你的磁场值计算函数是B(Z, rho)，这里用示例函数代替
B = np.exp(-rho_mesh/5) * np.cos(Z_mesh/2)

2. 创建新的Y坐标数组，把rho映射到Y的绝对值，Z的正负对应Y的正负：

# 当Z为负时，Y取-rho；Z为正时，Y取rho
Y_mesh = np.where(Z_mesh < 0, -rho_mesh, rho_mesh)

3. 绘制plot_surface，并自定义Y轴刻度标签：

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 使用新的Y_mesh绘图
ax.plot_surface(Z_mesh, Y_mesh, B, cmap='viridis')

# 设置Y轴刻度：从-10到10，刻度标签显示对应的rho值（绝对值）
yticks = np.arange(-10, 11, 1)
ax.set_yticks(yticks)
ax.set_yticklabels([abs(tick) for tick in yticks])

# 设置轴标签
ax.set_xlabel('Z')
ax.set_ylabel('P (rho)')
ax.set_zlabel('Magnetic Field')

plt.show()

这样处理后，Y轴的零点在中间，左侧刻度显示10到0，右侧显示0到10，完美匹配你的需求，同时所有Z正负对应的磁场数据都能正常显示。

方案2：仅调整Y轴刻度显示（不修改坐标） #

如果你不想改变原始的Y坐标（rho从0到10），只是希望刻度显示成10,9,...0,1,...的形式，可以自定义刻度位置和标签，但这种方法可能会让数据的空间位置看起来有点反直觉，因为实际Y轴还是0到10，只是标签反向后又正向：

# 这种方法需要扩展Y轴范围并复制数据，不如方案1直观，仅作参考
rho = np.linspace(0, 10, 50)
Z = np.linspace(-10, 10, 50)
Z_mesh, rho_mesh = np.meshgrid(Z, rho)
B = np.exp(-rho_mesh/5) * np.cos(Z_mesh/2)

# 复制rho数据到10-20区间，对应Z正的部分
rho_extended = np.concatenate([rho[::-1], rho[1:]])
Z_extended = np.concatenate([Z[Z<0], Z[Z>=0]])
Z_ext_mesh, rho_ext_mesh = np.meshgrid(Z_extended, rho_extended)
# 这里需要对应复制磁场数据，逻辑相对繁琐，不推荐

为什么你的原始绘图会显示不全？ #

大概率是因为Z有正负，但你只把Y轴（rho）设置在了0-10，而Z负对应的磁场数据没有被正确映射到可视化空间中，导致部分数据被隐藏。方案1通过把Y轴扩展到-10到10，并用正负区分Z的方向，就能完整展示所有数据了。

内容的提问来源于stack exchange，提问作者Wouter Adem