Axes vs pyplot interface comparison in Matplotlib - Hands-On Comparison

Axes vs pyplot interface comparison

📖 Scenario: You are working on a simple data visualization project. You want to understand two ways to create plots using matplotlib: the pyplot interface and the Axes interface. This will help you choose the best method for your future projects.

🎯 Goal: Create two line plots showing the same data: one using the pyplot interface and one using the Axes interface. Compare their outputs.

📋 What You'll Learn

Create a list of numbers called data with values [1, 3, 2, 4]

Create a figure and axis object using plt.subplots()

Plot the data using the Axes interface method ax.plot()

Plot the same data using the pyplot interface method plt.plot()

Display both plots separately

💡 Why This Matters

🌍 Real World

Data scientists and analysts often create visualizations to understand data. Knowing both the pyplot and Axes interfaces helps you choose the best way to build clear and flexible plots.

💼 Career

Understanding matplotlib's two main plotting interfaces is essential for data visualization tasks in data science, analytics, and research roles.

1

Create the data list

Create a list called data with these exact values: [1, 3, 2, 4].

Matplotlib

# Create the list called data with values [1, 3, 2, 4]
data = []

Hint

Use square brackets to create a list and separate numbers with commas.

2

Create figure and axis objects

Import matplotlib.pyplot as plt. Then create a figure and axis object using plt.subplots() and save them as fig and ax.

Matplotlib

import matplotlib.pyplot as plt

# Create figure and axis objects using plt.subplots()
fig, ax = None, None

Hint

Use import matplotlib.pyplot as plt and then fig, ax = plt.subplots().

3

Plot data using Axes interface

Use the ax.plot() method to plot the data list on the axis object ax.

Matplotlib

import matplotlib.pyplot as plt

data = [1, 3, 2, 4]

fig, ax = plt.subplots()

# Plot data using ax.plot()
ax.plot([])

Hint

Call ax.plot(data) to plot the list on the axis.

4

Plot data using pyplot interface and show plots

Plot the same data list using plt.plot(). Then call plt.show() twice: once after ax.plot() to show the Axes plot, and once after plt.plot() to show the pyplot plot.

Matplotlib

import matplotlib.pyplot as plt

data = [1, 3, 2, 4]

fig, ax = plt.subplots()

ax.plot(data)

# Show the Axes plot

# Plot data using plt.plot()

# Show the pyplot plot

Hint

Use plt.plot(data) to plot with pyplot and plt.show() to display each plot.

Practice

1. Which statement best describes the difference between the pyplot interface and the Axes interface in matplotlib?

easy

A. Axes is simpler and good for quick plots, while pyplot offers more control for complex figures.

B. pyplot is simpler and good for quick plots, while Axes offers more control for complex figures.

C. Both pyplot and Axes interfaces are equally simple and offer the same control.

D. pyplot is only used for 3D plots, while Axes is for 2D plots.

Solution

1. Step 1: Understand pyplot interface

   The pyplot interface is designed for quick and easy plotting with simple commands.

2. Step 2: Understand Axes interface

   The Axes interface provides detailed control over plot elements, useful for complex figures.

3. Final Answer:

   pyplot is simpler and good for quick plots, while Axes offers more control for complex figures. -> Option B

4. Quick Check:

   Interface complexity = pyplot simple, Axes detailed [OK]

Hint: Remember: pyplot = quick, Axes = control [OK]

Common Mistakes:

• Confusing which interface is simpler
• Thinking both interfaces have same control level
• Assuming pyplot is only for 3D plots

2. Which of the following is the correct way to create a figure and axes using the Axes interface?

easy

A. fig, ax = plt.subplots()

B. ax = plt.figure()

C. fig = plt.axes()

D. plt.plot(), ax = plt.figure()

Solution

1. Step 1: Recall syntax for creating figure and axes

   The correct syntax to create a figure and axes object is fig, ax = plt.subplots().

2. Step 2: Check other options

   plt.figure() creates only a figure, plt.axes() creates axes but not figure, and plt.plot() is for plotting, not creating axes.

3. Final Answer:

   fig, ax = plt.subplots() -> Option A

4. Quick Check:

   Correct Axes creation = plt.subplots() [OK]

Hint: Use plt.subplots() to get figure and axes together [OK]

Common Mistakes:

• Using plt.figure() expecting axes returned
• Confusing plt.axes() with plt.figure()
• Trying to assign plt.plot() to axes

3. What will be the output of this code snippet?

import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.plot([1, 2, 3], [4, 5, 6])
plt.title('My Plot')
plt.show()

medium

A. A plot without title because plt.title() affects only pyplot interface.

B. An error because plt.title() cannot be used with Axes interface.

C. A blank figure with no plot or title.

D. A plot with line connecting points (1,4), (2,5), (3,6) and title 'My Plot'.

Solution

1. Step 1: Understand plotting with Axes

   The ax.plot() draws the line on the axes object.

2. Step 2: Understand plt.title() effect

   plt.title() sets the title on the current axes, so it works here.

3. Final Answer:

   A plot with line connecting points (1,4), (2,5), (3,6) and title 'My Plot'. -> Option D

4. Quick Check:

   Axes plot + plt.title() = plot with title [OK]

Hint: plt.title() sets title on current axes even with Axes interface [OK]

Common Mistakes:

• Thinking plt.title() won't work with Axes interface
• Expecting error due to mixing interfaces
• Assuming no plot will show

4. Identify the error in this code using the Axes interface:

import matplotlib.pyplot as plt
fig, ax = plt.subplots()
plt.figure()
plt.plot([1, 2, 3], [4, 5, 6])
ax.set_title('Test Plot')
plt.show()

medium

A. fig, ax = plt.subplots() is incorrect syntax.

B. ax.set_title() is invalid; should use plt.title() instead.

C. Using plt.plot() instead of ax.plot() causes the plot to appear on a different axes.

D. plt.show() is missing and causes no output.

Solution

1. Step 1: Check plotting method

   plt.plot() plots on the current pyplot axes, not the ax created by plt.subplots().

2. Step 2: Check title setting

   ax.set_title() correctly sets the title on the ax axes.

3. Final Answer:

   Using plt.plot() instead of ax.plot() causes the plot to appear on a different axes. -> Option C

4. Quick Check:

   Plotting on wrong axes = plt.plot() vs ax.plot() [OK]

Hint: Use ax.plot() to plot on specific axes, not plt.plot() [OK]

Common Mistakes:

• Confusing plt.plot() and ax.plot()
• Thinking ax.set_title() is invalid
• Missing plt.show() causing no output

5. You want to create a figure with two side-by-side plots sharing the same y-axis. Which approach using the Axes interface is best?

hard

A. Use fig, (ax1, ax2) = plt.subplots(1, 2, sharey=True) and plot on ax1 and ax2.

B. Call plt.plot() twice and then set plt.sharey(True).

C. Create two separate figures with plt.figure() and plot independently.

D. Use fig, ax = plt.subplots() and plot both lines on the same axes.

Solution

1. Step 1: Understand subplot creation with shared axis

   plt.subplots(1, 2, sharey=True) creates two axes side-by-side sharing the y-axis.

2. Step 2: Plot on each axes separately

   Plotting on ax1 and ax2 allows independent x-data but shared y-axis.

3. Final Answer:

   Use fig, (ax1, ax2) = plt.subplots(1, 2, sharey=True) and plot on ax1 and ax2. -> Option A

4. Quick Check:

   Shared y-axis subplots = plt.subplots with sharey=True [OK]

Hint: Use plt.subplots with sharey=True for shared y-axis plots [OK]

Common Mistakes:

• Trying to share axis with plt.plot() only
• Creating separate figures instead of subplots
• Plotting both lines on one axes when separate plots needed