求助：如何创建mesh grid以生成numpy plt.contour等高线图？

Let's get this sorted out—creating the right mesh grid is all about matching your matrix's dimensions, and I'll break down exactly how to do it without errors. Here's a complete, working code flow tailored to your case:

Step 1: Load Your Matrix Correctly #

First, make sure you're loading the TXT file as a 2D NumPy array (since contour plots require 2D data). Use numpy.loadtxt for this:

import numpy as np

# Replace 'your_matrix.txt' with your actual file path
matrix = np.loadtxt('your_matrix.txt')
# Double-check the matrix shape (rows, columns) to avoid surprises
print(f"Matrix shape: {matrix.shape}")

Step 2: Create the Matching Mesh Grid #

The common mistake with arange is not aligning the grid dimensions to your matrix's rows and columns. Here's how to generate X and Y grids that perfectly match your matrix:

# Get the number of rows and columns from your matrix
n_rows, n_cols = matrix.shape

# Create 1D arrays for x (columns) and y (rows) indices
x = np.arange(n_cols)  # X axis spans from 0 to (number of columns - 1)
y = np.arange(n_rows)  # Y axis spans from 0 to (number of rows - 1)

# Generate the 2D mesh grid
X, Y = np.meshgrid(x, y)

This ensures X and Y have the same shape as your matrix, which is required for plt.contour or plt.contourf to work without dimension mismatch errors.

Step 3: Plot the Contour Map #

Now you can safely generate your contour plot with Matplotlib:

import matplotlib.pyplot as plt

plt.figure(figsize=(10, 8))
# Use contourf for filled contours (contour for line-only)
contour = plt.contourf(X, Y, matrix, cmap='viridis', levels=20)  # Adjust levels for more/less detail

# Add a colorbar to show the 0-255 value scale
plt.colorbar(contour, label='Pixel Value (0-255)')

# Add labels and title for clarity
plt.xlabel('Column Index')
plt.ylabel('Row Index')
plt.title('Contour Plot of Your 0-255 Matrix')

plt.show()

Why Your Previous arange Attempt May Have Failed #

If you got errors before, it's likely because:

• You used arange with the total number of elements in the matrix instead of separate values for rows and columns
• The resulting X/Y grids had a different shape than your matrix (e.g., 1D instead of 2D, or mismatched rows/columns)

This code should resolve those issues—just swap in your actual file path and run it!

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