How to visualize scalar 2D data with Matplotlib?

To visualize scalar 2D data with Matplotlib, we create a pseudocolor plot that maps scalar values to colors across a 2D grid. This technique is useful for displaying functions of two variables, heatmaps, or any data that varies across a plane.

Basic Steps

The process involves these key steps:

• Create coordinate grids using np.meshgrid()
• Generate or compute scalar values for each grid point
• Use plt.pcolormesh() to create the visualization
• Apply colormaps to enhance data interpretation

Example: Visualizing a Mathematical Function

Here's how to create a pseudocolor plot of a 2D sinc function ?

import numpy as np
from matplotlib import pyplot as plt

# Set figure size
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

# Create coordinate arrays
n = 256
x = np.linspace(-3., 3., n)
y = np.linspace(-3., 3., n)

# Create coordinate grids
X, Y = np.meshgrid(x, y)

# Calculate scalar field
Z = X * np.sinc(X ** 2 + Y ** 2)

# Create pseudocolor plot
plt.pcolormesh(X, Y, Z, cmap='copper', shading='auto')
plt.colorbar(label='Z values')
plt.title('2D Sinc Function Visualization')
plt.xlabel('X')
plt.ylabel('Y')

plt.show()

Key Components

Coordinate Grids

The np.meshgrid() function creates 2D coordinate arrays from 1D arrays ?

import numpy as np

x = np.linspace(-2, 2, 5)
y = np.linspace(-1, 1, 3)
X, Y = np.meshgrid(x, y)

print("X grid:")
print(X)
print("\nY grid:")
print(Y)

X grid:
[[-2. -1.  0.  1.  2.]
 [-2. -1.  0.  1.  2.]
 [-2. -1.  0.  1.  2.]]

Y grid:
[[-1. -1. -1. -1. -1.]
 [ 0.  0.  0.  0.  0.]
 [ 1.  1.  1.  1.  1.]]

Colormap Options

Different colormaps provide various visual representations ?

import numpy as np
import matplotlib.pyplot as plt

# Create sample data
x = np.linspace(-2, 2, 20)
y = np.linspace(-2, 2, 20)
X, Y = np.meshgrid(x, y)
Z = np.sqrt(X**2 + Y**2)

# Create subplots with different colormaps
fig, axes = plt.subplots(1, 3, figsize=(12, 4))

# Viridis colormap
im1 = axes[0].pcolormesh(X, Y, Z, cmap='viridis', shading='auto')
axes[0].set_title('Viridis')
plt.colorbar(im1, ax=axes[0])

# Plasma colormap
im2 = axes[1].pcolormesh(X, Y, Z, cmap='plasma', shading='auto')
axes[1].set_title('Plasma')
plt.colorbar(im2, ax=axes[1])

# Coolwarm colormap
im3 = axes[2].pcolormesh(X, Y, Z, cmap='coolwarm', shading='auto')
axes[2].set_title('Coolwarm')
plt.colorbar(im3, ax=axes[2])

plt.tight_layout()
plt.show()

Advanced Techniques

Contour Overlay

Combine pseudocolor plots with contour lines for enhanced visualization ?

import numpy as np
import matplotlib.pyplot as plt

# Generate data
x = np.linspace(-3, 3, 100)
y = np.linspace(-3, 3, 100)
X, Y = np.meshgrid(x, y)
Z = np.exp(-(X**2 + Y**2)/2)

# Create combined plot
plt.figure(figsize=(8, 6))
plt.pcolormesh(X, Y, Z, cmap='Blues', shading='auto', alpha=0.8)
contours = plt.contour(X, Y, Z, colors='black', alpha=0.6, linewidths=0.8)
plt.clabel(contours, inline=True, fontsize=8)

plt.colorbar(label='Gaussian values')
plt.title('2D Gaussian with Contour Lines')
plt.xlabel('X')
plt.ylabel('Y')
plt.show()

Comparison of Visualization Methods

Conclusion

Matplotlib's pcolormesh() provides an effective way to visualize scalar 2D data using pseudocolors. Combine it with appropriate colormaps and additional elements like contour lines to create informative scientific visualizations that clearly represent your data patterns.