3D wireframe plots in Matplotlib - Mini Project: Build & Apply

3D Wireframe Plots

📖 Scenario: You are a data scientist exploring how to visualize 3D surfaces. Wireframe plots help you see the shape of a surface using lines instead of solid colors. This is useful when you want to understand the structure of data in three dimensions.

🎯 Goal: You will create a 3D wireframe plot using matplotlib. You will first set up the data points, then configure the grid, apply the wireframe plotting, and finally display the plot.

📋 What You'll Learn

Create 1D arrays for X and Y coordinates using numpy

Create a meshgrid from X and Y arrays

Calculate Z values using a mathematical function

Use matplotlib to create a 3D wireframe plot

Display the plot with labels

💡 Why This Matters

🌍 Real World

3D wireframe plots are used in engineering and science to visualize surfaces like terrain, temperature, or pressure distributions.

💼 Career

Data scientists and analysts use 3D plots to understand complex data and communicate insights visually.

1

Create X and Y coordinate arrays

Import numpy as np. Create a 1D numpy array called x with values from 0 to 5 (inclusive) with 6 points. Create another 1D numpy array called y with values from 0 to 5 (inclusive) with 6 points.

Matplotlib

# Your code here

Hint

Use np.linspace(start, stop, num_points) to create evenly spaced values.

2

Create meshgrid for X and Y

Use np.meshgrid with x and y to create 2D arrays called X and Y.

Matplotlib

import numpy as np
x = np.linspace(0, 5, 6)
y = np.linspace(0, 5, 6)
# Your code here

Hint

Use X, Y = np.meshgrid(x, y) to create coordinate matrices.

3

Calculate Z values for the surface

Create a 2D array Z by calculating Z = np.sin(X) + np.cos(Y).

Matplotlib

import numpy as np
x = np.linspace(0, 5, 6)
y = np.linspace(0, 5, 6)
X, Y = np.meshgrid(x, y)
# Your code here

Hint

Use numpy's sin and cos functions on the meshgrid arrays.

4

Create and display the 3D wireframe plot

Import matplotlib.pyplot as plt and Axes3D from mpl_toolkits.mplot3d. Create a figure and add a 3D subplot. Use ax.plot_wireframe(X, Y, Z) to plot the wireframe. Set the X, Y, and Z axis labels to "X axis", "Y axis", and "Z axis" respectively. Finally, use plt.show() to display the plot.

Matplotlib

import numpy as np
x = np.linspace(0, 5, 6)
y = np.linspace(0, 5, 6)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) + np.cos(Y)
# Your code here

Hint

Use fig = plt.figure() and ax = fig.add_subplot(111, projection='3d') to create a 3D plot. Then use ax.plot_wireframe(X, Y, Z) to draw the wireframe.

Practice

1. What does a 3D wireframe plot in matplotlib primarily show?

easy

A. Only the color distribution of data points

B. A flat 2D scatter plot

C. The shape of data or functions in three dimensions using lines

D. A pie chart with 3D effects

Solution

1. Step 1: Understand the purpose of 3D wireframe plots

   3D wireframe plots use a grid of lines to represent the shape of data or functions in three dimensions.

2. Step 2: Compare with other plot types

   Unlike scatter or pie charts, wireframe plots focus on the surface structure, not just colors or flat points.

3. Final Answer:

   The shape of data or functions in three dimensions using lines -> Option C

4. Quick Check:

   3D wireframe = 3D shape with lines [OK]

Hint: Wireframe plots show 3D shapes with lines, not colors or points [OK]

Common Mistakes:

• Confusing wireframe with scatter or surface plots
• Thinking wireframe shows only colors
• Assuming wireframe is 2D

2. Which of the following is the correct way to create a 3D wireframe plot using matplotlib?

easy

A. ax.plot_wireframe(X, Y, Z)

B. ax.plot_surface(X, Y, Z)

C. plt.plot_wireframe(X, Y, Z)

D. ax.scatter_wireframe(X, Y, Z)

Solution

1. Step 1: Identify the correct method for wireframe plots

   The method plot_wireframe is called on the 3D axes object ax.

2. Step 2: Eliminate incorrect options

   plot_surface creates a surface plot, not wireframe. plt.plot_wireframe is invalid because plt does not have this method. scatter_wireframe does not exist.

3. Final Answer:

   ax.plot_wireframe(X, Y, Z) -> Option A

4. Quick Check:

   Wireframe method is plot_wireframe on ax [OK]

Hint: Use ax.plot_wireframe for 3D wireframe plots [OK]

Common Mistakes:

• Calling plot_wireframe on plt instead of ax
• Using plot_surface instead of plot_wireframe
• Using non-existent methods like scatter_wireframe

3. What will the following code output?

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
X = np.arange(-5, 6, 5)
Y = np.arange(-5, 6, 5)
X, Y = np.meshgrid(X, Y)
Z = X**2 - Y**2
ax.plot_wireframe(X, Y, Z, rstride=1, cstride=1)
plt.show()

medium

A. A 3D wireframe plot showing a saddle shape

B. A 2D line plot of X and Y

C. A scatter plot of points

D. An error due to incorrect meshgrid usage

Solution

1. Step 1: Understand the meshgrid and function

   X and Y create a grid from -5 to 5 with step 5, so points at -5, 0, 5. Z = X^2 - Y^2 forms a saddle shape.

2. Step 2: Analyze the plot_wireframe call

   Using rstride=1 and cstride=1 plots all grid lines, producing a wireframe of the saddle surface.

3. Final Answer:

   A 3D wireframe plot showing a saddle shape -> Option A

4. Quick Check:

   Wireframe of Z = X^2 - Y^2 = saddle shape [OK]

Hint: Z = X² - Y² creates a saddle; wireframe shows surface shape [OK]

Common Mistakes:

• Thinking meshgrid creates error
• Confusing wireframe with scatter or 2D plot
• Ignoring the shape of Z function

4. Identify the error in this code snippet for a 3D wireframe plot:

import numpy as np
import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
X = np.linspace(-3, 3, 10)
Y = np.linspace(-3, 3, 10)
Z = np.sin(X) * np.cos(Y)
ax.plot_wireframe(X, Y, Z)
plt.show()

medium

A. X and Y should be lists, not arrays

B. Missing import for Axes3D

C. plot_wireframe does not exist

D. Z is not a 2D array matching X and Y meshgrid shape

Solution

1. Step 1: Check shapes of X, Y, and Z

   X and Y are 1D arrays; Z is computed element-wise but is also 1D, not 2D grid.

2. Step 2: Understand plot_wireframe requirements

   plot_wireframe requires X, Y, Z to be 2D arrays from meshgrid to plot a surface grid.

3. Final Answer:

   Z is not a 2D array matching X and Y meshgrid shape -> Option D

4. Quick Check:

   plot_wireframe needs 2D X, Y, Z arrays [OK]

Hint: Use meshgrid to make X, Y, Z 2D arrays for wireframe [OK]

Common Mistakes:

• Passing 1D arrays instead of meshgrid 2D arrays
• Ignoring shape mismatch errors
• Assuming plot_wireframe works with 1D inputs

5. You want to plot a 3D wireframe of the function Z = sin(sqrt(X² + Y²)) over the range -6 to 6 for both X and Y with a grid spacing of 0.5. Which code snippet correctly creates this plot with a blue wireframe and stride of 5?

hard

A. import numpy as np
import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
X = np.linspace(-6, 6, 25)
Y = np.linspace(-6, 6, 25)
Z = np.sin(np.sqrt(X**2 + Y**2))
ax.plot_wireframe(X, Y, Z, color='blue', stride=5)
plt.show()

B. import numpy as np
import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
X = np.arange(-6, 6.5, 0.5)
Y = np.arange(-6, 6.5, 0.5)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
ax.plot_wireframe(X, Y, Z, rstride=5, cstride=5, color='blue')
plt.show()

C. import numpy as np
import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
X = np.arange(-6, 6, 0.5)
Y = np.arange(-6, 6, 0.5)
X, Y = np.meshgrid(X, Y)
Z = np.sin(np.sqrt(X**2 + Y**2))
ax.plot_wireframe(X, Y, Z, rstride=5, cstride=5, color='red')
plt.show()

D. import numpy as np
import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
X = np.arange(-6, 6, 0.5)
Y = np.arange(-6, 6, 0.5)
X, Y = np.meshgrid(X, Y)
Z = np.sin(np.sqrt(X**2 + Y**2))
ax.plot_wireframe(X, Y, Z, rstride=0.5, cstride=0.5, color='blue')
plt.show()

Solution

1. Step 1: Create X and Y grids with correct range and spacing

   Using np.arange(-6, 6.5, 0.5) ensures points from -6 to 6 with 0.5 spacing. Then meshgrid creates 2D arrays.

2. Step 2: Calculate Z and plot with correct stride and color

   Z is computed as sin(sqrt(X² + Y²)). The wireframe uses rstride=5 and cstride=5 for spacing lines, and color='blue' for blue lines.

3. Final Answer:

   Code snippet A correctly creates the desired 3D wireframe plot -> Option B

4. Quick Check:

   Correct meshgrid, stride=5, color='blue' = import numpy as np import matplotlib.pyplot as plt fig = plt.figure() ax = fig.add_subplot(111, projection='3d') X = np.arange(-6, 6.5, 0.5) Y = np.arange(-6, 6.5, 0.5) X, Y = np.meshgrid(X, Y) R = np.sqrt(X**2 + Y**2) Z = np.sin(R) ax.plot_wireframe(X, Y, Z, rstride=5, cstride=5, color='blue') plt.show() [OK]

Hint: Use meshgrid, rstride/cstride for spacing, color param for wireframe [OK]

Common Mistakes:

• Using stride instead of rstride and cstride
• Incorrect range or missing meshgrid
• Wrong color or stride values