Unequal subplot sizes in Matplotlib - Time & Space Complexity

Time Complexity: Unequal subplot sizes

O(n)

Understanding Time Complexity

We want to understand how the time to create plots changes when we use subplots of different sizes in matplotlib.

How does the number and size of subplots affect the work matplotlib does?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import matplotlib.pyplot as plt

fig = plt.figure()
ax1 = fig.add_axes([0.1, 0.1, 0.6, 0.8])  # Large subplot
ax2 = fig.add_axes([0.75, 0.1, 0.2, 0.3]) # Smaller subplot

ax1.plot(range(100))
ax2.plot(range(50))
plt.show()
    

This code creates a figure with two subplots of different sizes and plots data on each.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Plotting data points on each subplot.
• How many times: Once per data point in each subplot (100 points in first, 50 in second).

How Execution Grows With Input

As the number of points to plot increases, the work grows roughly in direct proportion.

Input Size (n)	Approx. Operations
10	About 10 operations per subplot
100	About 100 operations per subplot
1000	About 1000 operations per subplot

Pattern observation: Doubling the number of points roughly doubles the work needed to plot.

Final Time Complexity

Time Complexity: O(n)

This means the time to draw the plots grows linearly with the number of points plotted.

Common Mistake

[X] Wrong: "Making one subplot bigger will make the whole plotting time much slower regardless of data size."

[OK] Correct: The size of the subplot affects drawing area but the main time cost comes from how many data points are plotted, not the subplot size alone.

Interview Connect

Understanding how plotting time grows helps you explain performance in data visualization tasks clearly and confidently.

Self-Check

"What if we added a third subplot with twice as many points as the largest one? How would the time complexity change?"

Practice

1. What is the main advantage of using unequal subplot sizes in matplotlib?

easy

A. To automatically adjust colors of the plots

B. To make all charts the same size for uniformity

C. To highlight important charts by giving them more space

D. To reduce the total number of charts displayed

Solution

1. Step 1: Understand subplot size purpose

   Unequal subplot sizes allow some plots to be larger, drawing more attention.

2. Step 2: Compare options

   Only To highlight important charts by giving them more space mentions highlighting important charts by size, which is the main advantage.

3. Final Answer:

   To highlight important charts by giving them more space -> Option C

4. Quick Check:

   Unequal sizes highlight key charts [OK]

Hint: Unequal sizes = emphasize important plots [OK]

Common Mistakes:

• Thinking all subplots must be equal size
• Confusing size with color or number of plots
• Assuming unequal sizes reduce charts

2. Which of the following is the correct way to create subplots with unequal sizes using gridspec_kw in plt.subplots()?

easy

A. plt.subplots(1, 2, gridspec_kw={'width_ratios': [2, 1]})

B. plt.subplots(2, 1, gridspec_kw={'height_ratios': [1, 2]})

C. plt.subplots(1, 2, gridspec_kw={'height_ratios': [2, 1]})

D. plt.subplots(2, 1, gridspec_kw={'width_ratios': [1, 2]})

Solution

1. Step 1: Identify correct parameter for horizontal subplots

   For 1 row and 2 columns, use 'width_ratios' to set horizontal size ratios.

2. Step 2: Match ratios to subplot layout

   plt.subplots(1, 2, gridspec_kw={'width_ratios': [2, 1]}) uses 1 row, 2 columns with width_ratios, which is correct syntax.

3. Final Answer:

   plt.subplots(1, 2, gridspec_kw={'width_ratios': [2, 1]}) -> Option A

4. Quick Check:

   Width ratios for columns = plt.subplots(1, 2, gridspec_kw={'width_ratios': [2, 1]}) [OK]

Hint: Use width_ratios for columns, height_ratios for rows [OK]

Common Mistakes:

• Mixing width_ratios with rows instead of columns
• Using height_ratios for horizontal layout
• Wrong number of rows and columns in function

3. What will be the width ratio of the two subplots created by this code?

fig, axs = plt.subplots(1, 2, gridspec_kw={'width_ratios': [3, 1]})

medium

A. Both subplots have equal width

B. The subplots have random widths

C. The second subplot is 3 times wider than the first

D. The first subplot is 3 times wider than the second

Solution

1. Step 1: Understand width_ratios parameter

   The list [3, 1] means the first subplot width is 3 units, second is 1 unit.

2. Step 2: Interpret ratio meaning

   This means the first subplot is 3 times wider than the second subplot.

3. Final Answer:

   The first subplot is 3 times wider than the second -> Option D

4. Quick Check:

   Width ratio 3:1 means first is 3x wider [OK]

Hint: Ratio list directly shows relative sizes [OK]

Common Mistakes:

• Assuming equal widths despite ratios
• Reversing which subplot is wider
• Ignoring the ratio values

4. Identify the error in this code that tries to create unequal subplot sizes:

fig, axs = plt.subplots(2, 1, gridspec_kw={'width_ratios': [1, 2]})

medium

A. Incorrect number of subplots specified

B. Using width_ratios for vertical subplots instead of height_ratios

C. Missing plt.show() to display plots

D. width_ratios list length does not match subplot count

Solution

1. Step 1: Check subplot layout

   2 rows and 1 column means vertical stacking of subplots.

2. Step 2: Verify correct ratio parameter

   For vertical layout, use 'height_ratios', not 'width_ratios'.

3. Final Answer:

   Using width_ratios for vertical subplots instead of height_ratios -> Option B

4. Quick Check:

   Vertical layout needs height_ratios [OK]

Hint: Use height_ratios for rows, width_ratios for columns [OK]

Common Mistakes:

• Confusing width_ratios and height_ratios
• Ignoring subplot grid shape
• Assuming plt.show() affects subplot sizes

5. You want to create a figure with 3 subplots in one row where the middle subplot is twice as wide as the others. Which code correctly achieves this?

hard

A. plt.subplots(1, 3, gridspec_kw={'width_ratios': [1, 2, 1]})

B. plt.subplots(3, 1, gridspec_kw={'height_ratios': [1, 2, 1]})

C. plt.subplots(1, 3, gridspec_kw={'height_ratios': [1, 2, 1]})

D. plt.subplots(3, 1, gridspec_kw={'width_ratios': [1, 2, 1]})

Solution

1. Step 1: Determine subplot layout

   One row and three columns means horizontal layout.

2. Step 2: Choose correct ratio parameter

   For horizontal layout, use 'width_ratios' to set widths.

3. Step 3: Set ratios to make middle subplot twice as wide

   Use [1, 2, 1] so middle subplot is twice the width of others.

4. Final Answer:

   plt.subplots(1, 3, gridspec_kw={'width_ratios': [1, 2, 1]}) -> Option A

5. Quick Check:

   Horizontal layout with width_ratios = plt.subplots(1, 3, gridspec_kw={'width_ratios': [1, 2, 1]}) [OK]

Hint: Middle subplot twice wide = width_ratios [1, 2, 1] [OK]

Common Mistakes:

• Using height_ratios for horizontal layout
• Setting wrong number of rows and columns
• Reversing ratio order