使用ax.plot_surface绘制3D曲面图遇异常：全黑且色条不匹配

Hey there! I’ve run into these exact headaches before when working with plot_surface, so let’s break down what’s likely going wrong and how to fix it step by step.

Common Causes & Fixes #

1. Your x, y, z data are 1D arrays (not 2D grids)
   ax.plot_surface relies on 2D arrays for x, y, and z—because it needs to map every combination of x and y coordinates to a corresponding z value. If you’re feeding in 1D lists/arrays, Matplotlib can’t properly render the surface, which often leads to a weird black plot or no plot at all.

Fix this using numpy.meshgrid to convert your 1D x and y data into 2D grid matrices. For example:

import numpy as np
# Your manual 1D input data (replace with your values)
x_1d = np.array([1, 2, 3, 4])
y_1d = np.array([5, 6, 7, 8])
# Convert to 2D grids
x_2d, y_2d = np.meshgrid(x_1d, y_1d)
# Now your z data needs to be a 2D array matching x_2d/y_2d's shape (4x4 here)
z_2d = np.array([
    [10, 12, 14, 16],
    [11, 13, 15, 17],
    [12, 14, 16, 18],
    [13, 15, 17, 19]
])

2. Missing colormap or invalid z value range
   A black plot usually happens when you don’t specify a colormap (cmap), or if all your z values are identical (so there’s no color variation to display). Even if z values vary, failing to link the color mapping to z will result in a flat, uninformative color.

3. Colorbar isn’t linked to the surface plot
   If your colorbar doesn’t match your z values, it’s almost certainly because you didn’t pass the returned Surface object from plot_surface to plt.colorbar(). The colorbar needs this reference to map colors correctly to your z data.

Full Working Example #

Here’s a corrected code snippet that addresses all these issues:

import numpy as np
import matplotlib.pyplot as plt

# Set up the figure and 3D axis
fig = plt.figure()
ax = fig.add_subplot(projection='3d')

# Your manual 1D x/y data (adjust to your actual values)
x_1d = np.array([0, 1, 2, 3])
y_1d = np.array([0, 1, 2, 3])

# Convert to 2D grids
x, y = np.meshgrid(x_1d, y_1d)

# Manual 2D z data (must match x/y's shape: 4x4 here)
z = np.array([
    [0, 1, 4, 9],
    [1, 2, 5, 10],
    [4, 5, 8, 13],
    [9, 10, 13, 18]
])

# Plot the surface, specify a colormap, and capture the Surface object
surf = ax.plot_surface(x, y, z, cmap='viridis', edgecolor='none')

# Add a colorbar linked directly to the surface plot
fig.colorbar(surf, ax=ax, shrink=0.5, aspect=5)

# Add labels for clarity
ax.set_xlabel('X Axis')
ax.set_ylabel('Y Axis')
ax.set_zlabel('Z Axis')
ax.set_title('Fixed 3D Surface Plot')

plt.show()

Quick Checks for Your Own Code #

• Verify that x, y, and z all have the same 2D shape (use print(x.shape, y.shape, z.shape) to confirm).
• Ensure your z values have meaningful variation (no all-identical entries).
• Always assign the result of plot_surface to a variable (like surf) and pass that variable to fig.colorbar().

内容的提问来源于stack exchange，提问作者J.Y.