Highlighting date ranges in Matplotlib - Time & Space Complexity

Time Complexity: Highlighting date ranges

O(n)

Understanding Time Complexity

When we highlight date ranges on a plot, we want to know how the time to draw changes as we add more highlights.

How does adding more date ranges affect the work matplotlib does?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import matplotlib.pyplot as plt
import pandas as pd

fig, ax = plt.subplots()
dates = pd.date_range('2024-01-01', periods=100)
values = range(100)
ax.plot(dates, values)

# Highlight 3 date ranges
for start, end in [("2024-01-10", "2024-01-15"), ("2024-01-30", "2024-02-05"), ("2024-02-20", "2024-02-25")]:
    ax.axvspan(pd.to_datetime(start), pd.to_datetime(end), color='yellow', alpha=0.3)
plt.show()

This code plots 100 dates and highlights three separate date ranges with colored spans.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Loop over the list of date ranges to add highlight spans.
• How many times: Once per highlighted date range (3 times in this example).

How Execution Grows With Input

Each additional date range adds one more highlight operation.

Input Size (number of highlights)	Approx. Operations
3	3 highlight calls
10	10 highlight calls
100	100 highlight calls

Pattern observation: The work grows directly with the number of highlighted date ranges.

Final Time Complexity

Time Complexity: O(n)

This means the time to add highlights grows linearly with how many date ranges you highlight.

Common Mistake

[X] Wrong: "Highlighting many date ranges happens instantly no matter how many."

[OK] Correct: Each highlight adds work, so more highlights take more time to draw.

Interview Connect

Understanding how adding visual elements affects performance helps you write efficient plotting code and explain your choices clearly.

Self-Check

What if we changed the code to highlight every single date point instead of ranges? How would the time complexity change?

Practice

1. What does the plt.axvspan(start_date, end_date) function do in matplotlib?

easy

A. It draws a horizontal line at start_date.

B. It highlights a vertical span (date range) on the plot between start_date and end_date.

C. It creates a scatter plot of dates between start_date and end_date.

D. It sets the x-axis limits to start_date and end_date.

Solution

1. Step 1: Understand the function purpose

   plt.axvspan is used to highlight a vertical area on the plot, usually to mark a range on the x-axis.

2. Step 2: Match parameters to behavior

   The parameters start_date and end_date define the start and end of this vertical span on the x-axis, highlighting that date range.

3. Final Answer:

   It highlights a vertical span (date range) on the plot between start_date and end_date. -> Option B

4. Quick Check:

   Highlight date range = plt.axvspan(start_date, end_date) [OK]

Hint: Remember axvspan highlights vertical areas between two x-values [OK]

Common Mistakes:

• Confusing axvspan with line drawing functions
• Thinking it changes axis limits
• Assuming it plots points instead of highlighting

2. Which of the following is the correct syntax to highlight the date range from '2023-01-01' to '2023-01-10' with a semi-transparent red color in matplotlib?

easy

A. plt.axvspan(start='2023-01-01', end='2023-01-10', color='red', alpha=0.5)

B. plt.axvspan('2023-01-01', '2023-01-10', color='red', transparency=0.5)

C. plt.axvspan('2023-01-01', '2023-01-10', color='red', alpha=0.5)

D. plt.axvspan('2023-01-01', '2023-01-10', alpha=50, color='red')

Solution

1. Step 1: Check parameter names and order

   plt.axvspan takes start and end positions as first two arguments, then keyword arguments like color and alpha.

2. Step 2: Validate correct keyword for transparency

   alpha controls transparency and should be between 0 and 1. The value 0.5 means 50% transparent.

3. Final Answer:

   plt.axvspan('2023-01-01', '2023-01-10', color='red', alpha=0.5) -> Option C

4. Quick Check:

   Use alpha for transparency, not transparency keyword [OK]

Hint: Use alpha (0-1) for transparency, not transparency keyword [OK]

Common Mistakes:

• Using 'transparency' instead of 'alpha'
• Passing alpha as 50 instead of 0.5
• Using named parameters start= and end= which are invalid

3. Given the following code, what will be the color and transparency of the highlighted date range?

import matplotlib.pyplot as plt
import pandas as pd

dates = pd.date_range('2023-01-01', periods=5)
values = [10, 20, 15, 25, 30]
plt.plot(dates, values)
plt.axvspan(pd.Timestamp('2023-01-02'), pd.Timestamp('2023-01-04'), color='green', alpha=0.3)
plt.show()

medium

A. A green shaded area with 30% opacity between Jan 2 and Jan 4

B. A green shaded area with 70% opacity between Jan 2 and Jan 4

C. A fully opaque green shaded area between Jan 2 and Jan 4

D. No shaded area will appear because alpha is too low

Solution

1. Step 1: Understand alpha parameter meaning

   Alpha=0.3 means 30% opacity (or 70% transparent), so the color is faint but visible.

2. Step 2: Confirm color and date range

   The color is green, and the highlighted range is from Jan 2 to Jan 4 as specified.

3. Final Answer:

   A green shaded area with 30% opacity between Jan 2 and Jan 4 -> Option A

4. Quick Check:

   alpha=0.3 means 30% opacity, color='green' [OK]

Hint: Alpha 0.3 means 30% opacity, not 70% [OK]

Common Mistakes:

• Confusing alpha as transparency percentage
• Thinking alpha=0.3 means 70% opacity
• Assuming no highlight appears if alpha is low

4. Identify the error in the following code snippet that tries to highlight a date range:

import matplotlib.pyplot as plt
plt.axvspan('2023-01-01', '2023-01-05', color='blue', alpha=0.4)
plt.show()

medium

A. Dates are passed as strings; they should be converted to datetime objects.

B. The alpha value must be an integer, not a float.

C. The color parameter should be 'colour' instead of 'color'.

D. plt.show() is missing and must be added.

Solution

1. Step 1: Check parameter types for axvspan

   plt.axvspan expects numeric or datetime values for start and end. Passing strings directly causes errors.

2. Step 2: Correct usage for dates

   Convert strings to datetime objects using pd.Timestamp or datetime before passing.

3. Final Answer:

   Dates are passed as strings; they should be converted to datetime objects. -> Option A

4. Quick Check:

   axvspan needs datetime, not string dates [OK]

Hint: Convert date strings to datetime objects before axvspan [OK]

Common Mistakes:

• Passing date strings directly without conversion
• Misusing alpha as integer
• Wrong color parameter spelling

5. You have a time series plot of daily sales data from '2023-01-01' to '2023-01-31'. You want to highlight weekends (Saturdays and Sundays) with a light gray transparent background. Which approach correctly applies this using matplotlib?

hard

A. Use plt.fill_between with weekend dates and color='gray', alpha=0.2.

B. Use plt.axvspan once with start='2023-01-01' and end='2023-01-31' and color='gray', alpha=0.2.

C. Plot vertical lines on weekends using plt.axvline with color='gray' and alpha=0.2.

D. Loop through dates, check if weekday is Saturday or Sunday, then use plt.axvspan for each weekend with color='gray' and alpha=0.2.

Solution

1. Step 1: Identify weekend dates in the range

   Loop through each date, check if it is Saturday (weekday=5) or Sunday (weekday=6).

2. Step 2: Highlight each weekend separately

   Use plt.axvspan for each weekend day or range with desired color and transparency.

3. Step 3: Understand why other options fail

   Use plt.axvspan once with start='2023-01-01' and end='2023-01-31' and color='gray', alpha=0.2. highlights entire month, not weekends. Plot vertical lines on weekends using plt.axvline with color='gray' and alpha=0.2. draws lines, not spans. Use plt.fill_between with weekend dates and color='gray', alpha=0.2. is for y-axis fills, not x-axis spans.

4. Final Answer:

   Loop through dates, check if weekday is Saturday or Sunday, then use plt.axvspan for each weekend with color='gray' and alpha=0.2. -> Option D

5. Quick Check:

   Highlight weekends individually with axvspan [OK]

Hint: Highlight each weekend separately with axvspan in a loop [OK]

Common Mistakes:

• Highlighting entire date range instead of weekends
• Using axvline instead of axvspan for spans
• Trying to fill y-axis area instead of x-axis date range