Why 3D visualization matters in Matplotlib - See It in Action

Why 3D visualization matters

📖 Scenario: Imagine you work in a company that studies how different factors affect sales. You have data with three features: advertising budget, price, and sales. You want to understand how these three things relate to each other. A 3D plot can help you see the relationship clearly.

🎯 Goal: You will create a simple 3D scatter plot using matplotlib to visualize how advertising budget and price relate to sales. This will show why 3D visualization matters when you have three variables.

📋 What You'll Learn

Create three lists named advertising, price, and sales with exact values

Create a variable fig for the figure and ax for 3D axes using matplotlib

Plot a 3D scatter plot using ax.scatter with the three lists

Label the axes with ax.set_xlabel, ax.set_ylabel, and ax.set_zlabel

Show the plot using plt.show()

💡 Why This Matters

🌍 Real World

3D visualization helps businesses and scientists see how three factors relate, like how advertising and price affect sales together.

💼 Career

Data scientists and analysts use 3D plots to explore and explain data with three variables, making insights clearer for decision makers.

1

Create the data lists

Create three lists called advertising, price, and sales with these exact values:
advertising = [100, 200, 300, 400, 500]
price = [10, 9, 8, 7, 6]
sales = [20, 40, 60, 80, 100]

Matplotlib

# Create the three lists advertising, price, and sales with the exact values
# Your code here

Hint

Use square brackets to create lists. Separate numbers with commas.

2

Set up the 3D plot

Import matplotlib.pyplot as plt and import Axes3D from mpl_toolkits.mplot3d. Then create a figure called fig using plt.figure() and create 3D axes called ax using fig.add_subplot(111, projection='3d').

Matplotlib

advertising = [100, 200, 300, 400, 500]
price = [10, 9, 8, 7, 6]
sales = [20, 40, 60, 80, 100]

# Import matplotlib.pyplot as plt and Axes3D
# Create fig with plt.figure()
# Create ax with fig.add_subplot(111, projection='3d')
# Your code here

Hint

Use import matplotlib.pyplot as plt and from mpl_toolkits.mplot3d import Axes3D. Then create the figure and 3D axes as shown.

3

Plot the 3D scatter plot

Use ax.scatter to plot the points with advertising on the x-axis, price on the y-axis, and sales on the z-axis. Then label the axes with ax.set_xlabel('Advertising Budget'), ax.set_ylabel('Price'), and ax.set_zlabel('Sales').

Matplotlib

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

advertising = [100, 200, 300, 400, 500]
price = [10, 9, 8, 7, 6]
sales = [20, 40, 60, 80, 100]

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Plot the 3D scatter plot with ax.scatter
# Label the axes with ax.set_xlabel, ax.set_ylabel, ax.set_zlabel
# Your code here

Hint

Use ax.scatter(x, y, z) to plot points in 3D. Then label each axis with ax.set_xlabel, ax.set_ylabel, and ax.set_zlabel.

4

Show the 3D plot

Use plt.show() to display the 3D scatter plot.

Matplotlib

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

advertising = [100, 200, 300, 400, 500]
price = [10, 9, 8, 7, 6]
sales = [20, 40, 60, 80, 100]

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

ax.scatter(advertising, price, sales)
ax.set_xlabel('Advertising Budget')
ax.set_ylabel('Price')
ax.set_zlabel('Sales')

# Show the plot with plt.show()
# Your code here

Hint

Use plt.show() to open the window with the 3D plot.

Practice

1. Why is 3D visualization important in data science?

easy

A. It helps show relationships among three variables clearly.

B. It makes data smaller and easier to store.

C. It removes the need for data cleaning.

D. It automatically finds patterns without analysis.

Solution

1. Step 1: Understand the role of 3D visualization

   3D visualization is used to display data with three variables or dimensions.

2. Step 2: Identify the benefit of 3D plots

   It helps reveal complex relationships that are hard to see in 2D plots.

3. Final Answer:

   It helps show relationships among three variables clearly. -> Option A

4. Quick Check:

   3D plots = show 3-variable relationships [OK]

Hint: 3D plots show three variables' relationships clearly [OK]

Common Mistakes:

• Thinking 3D plots reduce data size
• Believing 3D plots clean data automatically
• Assuming 3D plots find patterns without analysis

2. Which of the following is the correct way to import 3D plotting tools from matplotlib?

easy

A. import matplotlib3d as plt; from mpl_toolkits import Axes3D

B. from matplotlib import pyplot as plt; from mpl_toolkits.mplot3d import Axes3D

C. from matplotlib import Axes3D; import pyplot as plt

D. import matplotlib.pyplot as plt; import mpl_toolkits.mplot3d.Axes3D

Solution

1. Step 1: Recall correct import syntax for 3D plotting

   The standard way is to import pyplot as plt and import Axes3D from mpl_toolkits.mplot3d.

2. Step 2: Check each option for syntax correctness

   from matplotlib import pyplot as plt; from mpl_toolkits.mplot3d import Axes3D matches the correct syntax; others have wrong import statements or missing parts.

3. Final Answer:

   from matplotlib import pyplot as plt; from mpl_toolkits.mplot3d import Axes3D -> Option B

4. Quick Check:

   Correct 3D import = from matplotlib import pyplot as plt; from mpl_toolkits.mplot3d import Axes3D [OK]

Hint: Use 'from mpl_toolkits.mplot3d import Axes3D' for 3D plots [OK]

Common Mistakes:

• Using incorrect import paths
• Trying to import Axes3D directly from matplotlib
• Mixing import styles incorrectly

3. What will the following code output?

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter([1,2], [3,4], [5,6])
plt.show()

medium

A. Empty plot with no points shown

B. A 2D scatter plot ignoring the z values

C. SyntaxError due to wrong subplot code

D. A 3D scatter plot with points at (1,3,5) and (2,4,6)

Solution

1. Step 1: Analyze the code for 3D scatter plot creation

   The code creates a 3D subplot and plots points at given x, y, z coordinates.

2. Step 2: Understand the scatter method with 3D data

   ax.scatter plots points in 3D space at (1,3,5) and (2,4,6).

3. Final Answer:

   A 3D scatter plot with points at (1,3,5) and (2,4,6) -> Option D

4. Quick Check:

   3D scatter shows given points [OK]

Hint: ax.scatter with projection='3d' plots 3D points [OK]

Common Mistakes:

• Thinking it creates 2D plot ignoring z
• Expecting syntax error from subplot code
• Assuming plot is empty without points

4. Identify the error in this 3D plot code:

import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot([1,2,3], [4,5,6], [7,8,9])
plt.show()

medium

A. Missing projection='3d' in add_subplot

B. Using plt.figure() instead of plt.subplots()

C. plot() cannot take three lists as arguments

D. plt.show() is called before plotting

Solution

1. Step 1: Check subplot creation for 3D plotting

   To plot in 3D, add_subplot must include projection='3d'.

2. Step 2: Identify the cause of error

   Without projection='3d', ax.plot expects 2D data, so passing three lists causes error.

3. Final Answer:

   Missing projection='3d' in add_subplot -> Option A

4. Quick Check:

   3D plots need projection='3d' [OK]

Hint: Always add projection='3d' for 3D subplots [OK]

Common Mistakes:

• Forgetting projection='3d' in add_subplot
• Thinking plt.figure() is wrong here
• Believing plot() can't take three lists

5. You have a dataset with three features: height, weight, and age. How can 3D visualization help you understand this data better?

hard

A. By converting all features into a single number for easy plotting

B. By reducing the dataset to two features to simplify analysis

C. By plotting height, weight, and age on three axes to see their combined patterns

D. By ignoring age and focusing only on height and weight

Solution

1. Step 1: Understand the dataset features

   The dataset has three variables: height, weight, and age.

2. Step 2: Apply 3D visualization concept

   Plotting these three features on x, y, z axes helps see how they relate together.

3. Final Answer:

   By plotting height, weight, and age on three axes to see their combined patterns -> Option C

4. Quick Check:

   3D plot = visualize 3 features together [OK]

Hint: Use 3D plots to see three features' relationships [OK]

Common Mistakes:

• Reducing features loses important info
• Combining features into one number hides patterns
• Ignoring one feature misses data insights