3D wireframe plots in Matplotlib - Time & Space Complexity

Time Complexity: 3D wireframe plots

O(n^2)

Understanding Time Complexity

When creating 3D wireframe plots, we want to know how the time to draw the plot changes as we increase the data points.

We ask: How does the drawing time grow when we add more points to the 3D grid?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

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

x = np.linspace(-5, 5, 50)
y = np.linspace(-5, 5, 50)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.plot_wireframe(X, Y, Z)
plt.show()

This code creates a 3D wireframe plot using a grid of 50 by 50 points.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Plotting lines between points on the 2D grid to form the wireframe.
• How many times: For a grid of size n by n, lines are drawn along rows and columns, roughly 2 * n * (n - 1) times.

How Execution Grows With Input

As the grid size n increases, the number of lines to draw grows roughly with the square of n.

Input Size (n)	Approx. Operations
10	About 180 lines
100	About 19,800 lines
1000	About 1,998,000 lines

Pattern observation: Doubling n roughly quadruples the number of lines to draw, so the work grows fast as n grows.

Final Time Complexity

Time Complexity: O(n^2)

This means the time to draw the wireframe grows roughly with the square of the grid size.

Common Mistake

[X] Wrong: "The time to draw a 3D wireframe grows linearly with the number of points."

[OK] Correct: Because lines connect points in both directions (rows and columns), the number of lines grows with the square of the grid size, not just the number of points.

Interview Connect

Understanding how plotting time grows helps you explain performance in data visualization tasks and shows you can think about efficiency beyond just writing code.

Self-Check

"What if we changed the grid to be rectangular with dimensions n by m instead of n by n? How would the time complexity change?"

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