Matplotlib 3D曲面绘图异常求助：设置指定坐标值后仍显示全零图

嘿，我来帮你排查这个问题！你遇到的情况其实很常见——你的Z数组是512×512的全零矩阵，只把其中一个像素点Z[15,15]设为10，这个单点的高度在整个大尺寸的3D图里实在太不起眼了，就像在一片广阔的平地上放了一颗小石子，从全局视角看根本察觉不到它的存在。Matplotlib的3D绘图会自动适配整个数据集的数值范围，由于大部分区域都是0，这个单点的10就被“稀释”得几乎看不见了。

解决方案 #

这里给你几个可行的办法，帮你看到这个凸起：

1. 聚焦局部区域

通过限制X、Y轴的显示范围，放大到(15,15)附近的区域，同时手动设置Z轴范围来突出差异：

import matplotlib.pyplot as plt
import numpy as np

# Make data.
X = np.arange(0, 512, 1)
Y = np.arange(0, 512, 1)
X, Y = np.meshgrid(X, Y)
Z = np.zeros((512, 512), dtype=float)
Z[15,15] = 10

# Plot the surface.
fig = plt.figure(figsize=(6,6))
ax = fig.add_subplot(111, projection='3d')

# Plot a 3D surface
ax.plot_surface(X, Y, Z)

# 聚焦到目标点周围的小区域
ax.set_xlim(10, 20)
ax.set_ylim(10, 20)
# 手动设置Z轴范围，让0到10的差异更明显
ax.set_zlim(0, 11)

plt.show()

2. 扩大非零区域

如果不想局限于局部视角，可以把目标点周围的一片区域都设置为非零值，让凸起更显眼：

import matplotlib.pyplot as plt
import numpy as np

# Make data.
X = np.arange(0, 512, 1)
Y = np.arange(0, 512, 1)
X, Y = np.meshgrid(X, Y)
Z = np.zeros((512, 512), dtype=float)
# 将(15,15)周围3×3的区域都设为10
Z[14:17, 14:17] = 10

# Plot the surface.
fig = plt.figure(figsize=(6,6))
ax = fig.add_subplot(111, projection='3d')

# Plot a 3D surface
ax.plot_surface(X, Y, Z)

plt.show()

3. 调整颜色映射强化视觉差异

还可以搭配颜色映射来强化视觉差异，同时添加颜色条直观显示数值对应关系：

import matplotlib.pyplot as plt
import numpy as np

# Make data.
X = np.arange(0, 512, 1)
Y = np.arange(0, 512, 1)
X, Y = np.meshgrid(X, Y)
Z = np.zeros((512, 512), dtype=float)
Z[15,15] = 10

# Plot the surface.
fig = plt.figure(figsize=(8,6))
ax = fig.add_subplot(111, projection='3d')

# 添加颜色映射和颜色条
surf = ax.plot_surface(X, Y, Z, cmap='viridis')
fig.colorbar(surf, shrink=0.5, aspect=5)

# 聚焦局部区域
ax.set_xlim(10, 20)
ax.set_ylim(10, 20)
ax.set_zlim(0, 11)

plt.show()

这几种方法都能帮你看到那个被淹没的单点凸起，你可以根据自己的需求选择合适的方案~

内容的提问来源于stack exchange，提问作者Abhishek Ulhas.Hanchate