Annotating specific points in Matplotlib - Time & Space Complexity

Time Complexity: Annotating specific points

O(n)

Understanding Time Complexity

When we add annotations to specific points in a plot, it takes some extra work for the computer. We want to understand how this extra work grows as we add more points to annotate.

The question is: How does the time needed to add annotations change when we increase the number of points?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y = [2, 3, 5, 7, 11]

plt.scatter(x, y)

for i in range(len(x)):
    plt.annotate(f"({x[i]}, {y[i]})", (x[i], y[i]))

plt.show()

This code plots points and adds a text label next to each point showing its coordinates.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: The loop that adds an annotation for each point.
• How many times: Once for each point in the data (n times).

How Execution Grows With Input

Each point gets one annotation, so the work grows directly with the number of points.

Input Size (n)	Approx. Operations
10	10 annotations
100	100 annotations
1000	1000 annotations

Pattern observation: Doubling the points doubles the annotation work.

Final Time Complexity

Time Complexity: O(n)

This means the time to add annotations grows in a straight line with the number of points.

Common Mistake

[X] Wrong: "Adding annotations is instant and does not depend on how many points there are."

[OK] Correct: Each annotation requires extra drawing work, so more points mean more annotations and more time.

Interview Connect

Understanding how adding features like annotations affects performance helps you write better, more efficient code in real projects.

Self-Check

"What if we only annotate every 10th point instead of every point? How would the time complexity change?"

Practice

1. What is the main purpose of adding annotations to a matplotlib plot?

easy

A. To increase the size of the plot

B. To change the color of the plot lines

C. To add labels or notes to specific points on the plot

D. To remove grid lines from the plot

Solution

1. Step 1: Understand what annotations do

   Annotations are used to add extra information like labels or notes to specific points on a plot.

2. Step 2: Compare options with the purpose

   Only To add labels or notes to specific points on the plot describes adding labels or notes to points, which matches the purpose of annotations.

3. Final Answer:

   To add labels or notes to specific points on the plot -> Option C

4. Quick Check:

   Annotations = labels/notes on points [OK]

Hint: Annotations label points to explain data clearly [OK]

Common Mistakes:

• Confusing annotations with changing plot colors
• Thinking annotations resize the plot
• Assuming annotations remove grid lines

2. Which of the following is the correct syntax to add an annotation at point (2, 5) with text 'Peak' in matplotlib?

easy

A. plt.annotate('Peak', xy=(2, 5))

B. plt.annotation('Peak', (2, 5))

C. plt.annotate('Peak', point=(2, 5))

D. plt.annotate(text='Peak', position=(2, 5))

Solution

1. Step 1: Recall the correct function and parameters

   The correct function is plt.annotate with parameters: text first, then xy for the point coordinates.

2. Step 2: Check each option's syntax

   plt.annotate('Peak', xy=(2, 5)) uses plt.annotate('Peak', xy=(2, 5)) which is correct. Others use wrong function names or parameter names.

3. Final Answer:

   plt.annotate('Peak', xy=(2, 5)) -> Option A

4. Quick Check:

   Correct syntax uses xy= for point [OK]

Hint: Use plt.annotate(text, xy=(x,y)) to label points [OK]

Common Mistakes:

• Using plt.annotation instead of plt.annotate
• Wrong parameter names like point or position
• Swapping text and coordinate order

3. What will the following code display?

import matplotlib.pyplot as plt
x = [1, 2, 3]
y = [2, 5, 1]
plt.plot(x, y)
plt.annotate('High', xy=(2, 5), xytext=(3, 6), arrowprops=dict(facecolor='black'))
plt.show()

medium

A. A plot with a point at (3,6) labeled 'High' with an arrow pointing from (2,5)

B. Syntax error due to wrong arrowprops usage

C. A plot with no labels or arrows

D. A plot with a point at (2,5) labeled 'High' with an arrow pointing from (3,6)

Solution

1. Step 1: Understand plt.annotate parameters

   xy=(2,5) is the point to annotate, xytext=(3,6) is where the text is placed, arrowprops adds an arrow from text to point.

2. Step 2: Interpret the plot output

   The label 'High' appears at (3,6) with an arrow pointing to (2,5) on the plot line.

3. Final Answer:

   A plot with a point at (2,5) labeled 'High' with an arrow pointing from (3,6) -> Option D

4. Quick Check:

   xy = point, xytext = label position [OK]

Hint: xy=point, xytext=label position with arrow [OK]

Common Mistakes:

• Confusing xy and xytext roles
• Thinking arrow points from point to text
• Assuming no arrow appears without extra code

4. Identify the error in this code snippet that tries to annotate point (4, 7):

plt.plot([1,2,3,4], [3,5,2,7])
plt.annotate('Max', point=(4,7))
plt.show()

medium

A. plt.show() must come before plt.annotate

B. The parameter 'point' is invalid; should use 'xy' instead

C. plt.plot needs labels before annotation

D. The text 'Max' should be a number

Solution

1. Step 1: Check plt.annotate parameters

   The correct parameter for the point to annotate is 'xy', not 'point'.

2. Step 2: Identify the error cause

   Using 'point' causes a TypeError because plt.annotate does not recognize it.

3. Final Answer:

   The parameter 'point' is invalid; should use 'xy' instead -> Option B

4. Quick Check:

   Use xy= for annotation point [OK]

Hint: Use xy=, not point=, to specify annotation location [OK]

Common Mistakes:

• Using wrong parameter names like point or position
• Placing plt.show() before annotation
• Thinking annotation text must be numeric

5. You want to highlight the maximum value point on a plot with an annotation that places the label 10 units above the point and draws an arrow. Which code correctly achieves this for data x=[1,2,3], y=[4,9,7]?

hard

A. plt.plot(x, y)\nmax_idx = y.index(max(y))\nplt.annotate('Max', xy=(x[max_idx], y[max_idx]), xytext=(x[max_idx], y[max_idx]+10), arrowprops=dict(arrowstyle='->'))\nplt.show()

B. plt.plot(x, y)\nmax_idx = y.index(max(y))\nplt.annotate('Max', xytext=(x[max_idx], y[max_idx]), xy=(x[max_idx], y[max_idx]+10), arrowprops=dict(arrowstyle='->'))\nplt.show()

C. plt.plot(x, y)\nmax_val = max(y)\nplt.annotate('Max', xy=(max_val, x[max_idx]), xytext=(max_val, x[max_idx]+10), arrowprops=dict(arrowstyle='->'))\nplt.show()

D. plt.plot(x, y)\nmax_idx = y.index(max(y))\nplt.annotate('Max', xy=(x[max_idx], y[max_idx]), xytext=(x[max_idx], y[max_idx]-10), arrowprops=dict(arrowstyle='->'))\nplt.show()

Solution

1. Step 1: Find the max value index correctly

   max_idx = y.index(max(y)) finds the index of the highest y value.

2. Step 2: Set annotation point and label position

   xy is the max point (x[max_idx], y[max_idx]). xytext places label 10 units above: y[max_idx]+10.

3. Step 3: Use arrowprops to draw arrow

   arrowprops=dict(arrowstyle='->') correctly adds an arrow from label to point.

4. Final Answer:

   Code in option A correctly annotates max point with label 10 units above and arrow -> Option A

5. Quick Check:

   xy=point, xytext=above point, arrowprops=arrow [OK]

Hint: Use xytext=(x, y+offset) to place label above point [OK]

Common Mistakes:

• Swapping xy and xytext parameters
• Using wrong coordinates for annotation
• Placing label below instead of above
• Mixing x and y values in annotation