Custom legend entries in Matplotlib - Time & Space Complexity

Time Complexity: Custom legend entries

O(n)

Understanding Time Complexity

We want to understand how the time needed to create custom legend entries in matplotlib changes as the number of legend items grows.

How does adding more custom legend entries affect the drawing time?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import matplotlib.pyplot as plt
from matplotlib.lines import Line2D

custom_lines = [Line2D([0], [0], color='r', lw=4),
                Line2D([0], [0], color='g', lw=4)]
labels = ['Red line', 'Green line']

plt.legend(custom_lines, labels)
plt.show()

This code creates a legend with two custom line entries using matplotlib's Line2D objects.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Creating and adding each custom legend entry to the legend.
• How many times: Once per legend item, so the number of legend items (n) times.

How Execution Grows With Input

As the number of custom legend entries increases, the time to create and render the legend grows roughly in direct proportion.

Input Size (n)	Approx. Operations
10	About 10 legend entry creations and render steps
100	About 100 legend entry creations and render steps
1000	About 1000 legend entry creations and render steps

Pattern observation: The time grows linearly as the number of legend entries increases.

Final Time Complexity

Time Complexity: O(n)

This means the time to create the legend grows directly with the number of custom entries.

Common Mistake

[X] Wrong: "Adding more legend entries does not affect performance much because it's just a small part of the plot."

[OK] Correct: Each legend entry requires separate drawing and layout work, so more entries mean more work and longer rendering time.

Interview Connect

Understanding how UI elements like legends scale helps you reason about performance in data visualization tasks, a useful skill in many data science roles.

Self-Check

"What if we used a single legend entry with multiple labels instead of multiple separate entries? How would the time complexity change?"

Practice

1. What is the main purpose of creating custom legend entries in matplotlib?

easy

A. To change the plot background color

B. To increase the plot size

C. To add labels for plot elements that are not automatically labeled

D. To save the plot as an image file

Solution

1. Step 1: Understand legend purpose

   Legends explain what different plot elements represent.

2. Step 2: Identify custom legend role

   Custom legend entries add labels for elements not labeled automatically.

3. Final Answer:

   To add labels for plot elements that are not automatically labeled -> Option C

4. Quick Check:

   Custom legend entries = add labels [OK]

Hint: Custom legends label unlabeled plot parts quickly [OK]

Common Mistakes:

• Confusing legend with plot size or color
• Thinking legend changes plot background
• Assuming legend saves the plot

2. Which of the following is the correct way to create a custom legend entry for a red line using matplotlib.lines.Line2D?

easy

A. Line2D([0], [0], color='red', label='Red Line')

B. Line2D(color='red', label='Red Line')

C. Line2D([0,1], [0,1], color='red')

D. Line2D([0], [0], 'red', label='Red Line')

Solution

1. Step 1: Check Line2D constructor

   Line2D requires x and y data points as first two arguments.

2. Step 2: Verify correct syntax

   Line2D([0], [0], color='red', label='Red Line') uses [0], [0] for points, color='red', and label='Red Line' correctly.

3. Final Answer:

   Line2D([0], [0], color='red', label='Red Line') -> Option A

4. Quick Check:

   Line2D needs x,y points + color + label [OK]

Hint: Line2D needs x and y lists first, then style [OK]

Common Mistakes:

• Omitting x and y data points
• Passing color as positional argument
• Missing label parameter

3. What will be the output legend labels for this code snippet?

import matplotlib.pyplot as plt
from matplotlib.lines import Line2D

plt.plot([1, 2, 3], [4, 5, 6], 'b-')
custom_line = Line2D([0], [0], color='red', lw=2)
plt.legend([custom_line], ['Custom Red Line'])
plt.show()

medium

A. A legend with two labels: 'Custom Red Line' and 'blue line'

B. A legend with one label: 'blue line' shown in blue

C. No legend will appear

D. A legend with one label: 'Custom Red Line' shown in red

Solution

1. Step 1: Analyze plot and legend calls

   Plot draws a blue line but legend uses only custom_line handle with label 'Custom Red Line'.

2. Step 2: Understand legend content

   Legend shows only what is passed: one red line with label 'Custom Red Line'.

3. Final Answer:

   A legend with one label: 'Custom Red Line' shown in red -> Option D

4. Quick Check:

   Legend shows passed handles and labels only [OK]

Hint: Legend shows only handles and labels you pass [OK]

Common Mistakes:

• Assuming plot lines auto-appear in legend
• Expecting multiple legend entries without passing handles
• Thinking legend is empty

4. Identify the error in this code that tries to add a custom legend entry for a green square patch:

import matplotlib.pyplot as plt
from matplotlib.patches import Patch

patch = Patch(color='green', label='Green Square')
plt.legend(handles=patch, labels=['Green Square'])
plt.show()

medium

A. Patch cannot be used for legend entries

B. The 'handles' argument should be a list, not a single Patch object

C. The 'label' parameter is missing in Patch

D. plt.show() must come before plt.legend()

Solution

1. Step 1: Check handles argument type

   plt.legend expects handles as a list or iterable, not a single object.

2. Step 2: Identify fix

   Wrap patch in a list: handles=[patch] to fix the error.

3. Final Answer:

   The 'handles' argument should be a list, not a single Patch object -> Option B

4. Quick Check:

   Handles must be list or iterable [OK]

Hint: Always pass handles as a list to plt.legend [OK]

Common Mistakes:

• Passing single object instead of list to handles
• Ignoring label parameter in Patch
• Calling plt.show() before plt.legend()

5. You want to create a legend with a blue dashed line labeled 'Trend' and a red circle marker labeled 'Data Point'. Which code correctly creates custom legend entries for this?

hard

A. from matplotlib.lines import Line2D
legend_elements = [Line2D([0], [0], color='blue', lw=2, linestyle='--', label='Trend'), Line2D([0], [0], marker='o', color='red', linestyle='', label='Data Point')]
plt.legend(handles=legend_elements)

B. from matplotlib.patches import Patch
legend_elements = [Patch(color='blue', label='Trend'), Patch(color='red', label='Data Point')]
plt.legend(handles=legend_elements)

C. plt.legend(['Trend', 'Data Point'], loc='best')

D. plt.legend(['blue dashed line', 'red circle'])

Solution

1. Step 1: Identify correct legend handles

   Blue dashed line and red circle marker require Line2D objects with correct styles.

2. Step 2: Verify code correctness

   from matplotlib.lines import Line2D legend_elements = [Line2D([0], [0], color='blue', lw=2, linestyle='--', label='Trend'), Line2D([0], [0], marker='o', color='red', linestyle='', label='Data Point')] plt.legend(handles=legend_elements) creates Line2D with color='blue', linestyle='--' for 'Trend' and marker='o', color='red' for 'Data Point', then passes handles to plt.legend.

3. Final Answer:

   from matplotlib.lines import Line2D legend_elements = [Line2D([0], [0], color='blue', lw=2, linestyle='--', label='Trend'), Line2D([0], [0], marker='o', color='red', linestyle='', label='Data Point')] plt.legend(handles=legend_elements) -> Option A

4. Quick Check:

   Use Line2D with style and marker for custom legend [OK]

Hint: Use Line2D with marker and linestyle for custom legend [OK]

Common Mistakes:

• Using Patch for line and marker legend
• Passing only label strings without handles
• Ignoring linestyle or marker parameters