Dates on x-axis in Matplotlib - Time & Space Complexity

Time Complexity: Dates on x-axis

O(n)

Understanding Time Complexity

When plotting dates on the x-axis using matplotlib, it's important to understand how the plotting time changes as the number of dates grows.

We want to know how the time to draw the plot increases when we add more date points.

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import matplotlib.pyplot as plt
import matplotlib.dates as mdates
import datetime

dates = [datetime.date(2023, 1, i+1) for i in range(100)]
values = list(range(100))

plt.figure()
plt.plot(dates, values)
plt.gca().xaxis.set_major_formatter(mdates.DateFormatter('%Y-%m-%d'))
plt.gcf().autofmt_xdate()
plt.show()

This code plots 100 dates on the x-axis with corresponding values, formatting the dates for better readability.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Plotting each date-value pair on the graph.
• How many times: Once for each date in the list (n times).

How Execution Grows With Input

As the number of dates increases, the plotting time grows roughly in direct proportion.

Input Size (n)	Approx. Operations
10	10 plot points processed
100	100 plot points processed
1000	1000 plot points processed

Pattern observation: Doubling the number of dates roughly doubles the work done to plot them.

Final Time Complexity

Time Complexity: O(n)

This means the time to plot grows linearly with the number of dates you want to show.

Common Mistake

[X] Wrong: "Plotting dates is always slow because date formatting is complex and slows everything down."

[OK] Correct: While formatting dates adds some work, the main time cost comes from plotting each point, which grows linearly. The formatting cost is small compared to plotting many points.

Interview Connect

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

Self-Check

What if we changed the x-axis to show only monthly ticks instead of daily dates? How would the time complexity change?

Practice

1. What is the main reason to use datetime objects on the x-axis in matplotlib plots?

easy

A. To display dates properly and allow date-specific formatting

B. To speed up the plotting process

C. To change the color of the plot lines

D. To add grid lines automatically

Solution

1. Step 1: Understand the role of datetime objects

   Datetime objects represent dates and times in Python, allowing matplotlib to recognize and format them correctly on the x-axis.

2. Step 2: Identify the benefit for plotting

   Using datetime objects enables proper date formatting, such as showing months, days, or years, and helps in handling date ticks automatically.

3. Final Answer:

   To display dates properly and allow date-specific formatting -> Option A

4. Quick Check:

   Datetime objects = Proper date display [OK]

Hint: Use datetime objects to get correct date formatting on x-axis [OK]

Common Mistakes:

• Using strings instead of datetime objects for dates
• Expecting automatic color changes from datetime
• Confusing grid lines with date formatting

2. Which of the following is the correct way to import the module used for date formatting in matplotlib?

easy

A. import matplotlib.date as mdate

B. import matplotlib.dates as mdates

C. import matplotlib.datetime as mdates

D. import matplotlib.dateformatter as mdates

Solution

1. Step 1: Recall the correct module name

   The correct module for date formatting in matplotlib is called dates, not date or datetime.

2. Step 2: Match the import alias

   Common practice is to import it as mdates for convenience.

3. Final Answer:

   import matplotlib.dates as mdates -> Option B

4. Quick Check:

   matplotlib.dates = mdates [OK]

Hint: Remember: 'dates' plural, alias 'mdates' [OK]

Common Mistakes:

• Using singular 'date' instead of 'dates'
• Confusing datetime module with matplotlib.dates
• Using wrong alias causing import errors

3. What will the following code output?

import matplotlib.pyplot as plt
import matplotlib.dates as mdates
from datetime import datetime

dates = [datetime(2024, 1, 1), datetime(2024, 1, 2), datetime(2024, 1, 3)]
values = [10, 20, 15]

plt.plot(dates, values)
plt.gcf().autofmt_xdate()
plt.show()

medium

A. A line plot with dates on x-axis formatted and rotated for readability

B. A line plot with numeric x-axis values instead of dates

C. An error because dates are not converted to strings

D. A scatter plot with dates on y-axis

Solution

1. Step 1: Understand the plot function with datetime

   The plt.plot function accepts datetime objects on the x-axis and plots the corresponding values.

2. Step 2: Effect of autofmt_xdate()

   This function rotates and formats the date labels on the x-axis to prevent overlap and improve readability.

3. Final Answer:

   A line plot with dates on x-axis formatted and rotated for readability -> Option A

4. Quick Check:

   autofmt_xdate() rotates date labels [OK]

Hint: Use autofmt_xdate() to rotate date labels automatically [OK]

Common Mistakes:

• Expecting string conversion of dates automatically
• Confusing plot type with scatter
• Ignoring the effect of autofmt_xdate()

4. Identify the error in this code snippet that tries to plot dates on the x-axis:

import matplotlib.pyplot as plt
from datetime import datetime

dates = ['2024-01-01', '2024-01-02', '2024-01-03']
values = [5, 10, 15]

plt.plot(dates, values)
plt.show()

medium

A. plt.plot cannot plot lists

B. Values list length does not match dates list length

C. Missing import for matplotlib.dates module

D. Dates are strings, not datetime objects, causing incorrect x-axis formatting

Solution

1. Step 1: Check the type of dates list

   The dates are given as strings, which matplotlib does not automatically convert to dates for axis formatting.

2. Step 2: Understand the impact on plotting

   Using strings causes matplotlib to treat them as categorical labels, not dates, so date formatting and scaling won't work properly.

3. Final Answer:

   Dates are strings, not datetime objects, causing incorrect x-axis formatting -> Option D

4. Quick Check:

   Use datetime objects, not strings for dates [OK]

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

Common Mistakes:

• Using strings directly for dates
• Assuming matplotlib auto-converts strings to dates
• Ignoring the need for datetime conversion

5. You want to plot daily sales data for January 2024 with dates on the x-axis. You have a list of date strings and sales numbers. Which code snippet correctly converts the dates and formats the x-axis for clear display?

dates = ['2024-01-01', '2024-01-02', '2024-01-03']
sales = [100, 150, 120]

hard

A. from datetime import datetime
plt.plot([datetime(d) for d in dates], sales)
plt.show()

B. import matplotlib.pyplot as plt
plt.plot(dates, sales)
plt.xticks(rotation=45)
plt.show()

C. from datetime import datetime
import matplotlib.pyplot as plt
import matplotlib.dates as mdates

converted_dates = [datetime.strptime(d, '%Y-%m-%d') for d in dates]
plt.plot(converted_dates, sales)
plt.gca().xaxis.set_major_formatter(mdates.DateFormatter('%b %d'))
plt.gcf().autofmt_xdate()
plt.show()

D. import matplotlib.dates as mdates
plt.plot(dates, sales)
plt.gca().xaxis.set_major_locator(mdates.DayLocator())
plt.show()

Solution

1. Step 1: Convert date strings to datetime objects

   Use datetime.strptime with the correct format to convert each string to a datetime object.

2. Step 2: Format x-axis dates and rotate labels

   Use mdates.DateFormatter to set a readable date format and autofmt_xdate() to rotate labels for clarity.

3. Final Answer:

   The code snippet that correctly converts dates and formats x-axis -> Option C

4. Quick Check:

   Convert dates + format + rotate labels = from datetime import datetime import matplotlib.pyplot as plt import matplotlib.dates as mdates converted_dates = [datetime.strptime(d, '%Y-%m-%d') for d in dates] plt.plot(converted_dates, sales) plt.gca().xaxis.set_major_formatter(mdates.DateFormatter('%b %d')) plt.gcf().autofmt_xdate() plt.show() [OK]

Hint: Convert strings with strptime, format with DateFormatter, rotate labels [OK]

Common Mistakes:

• Plotting strings without conversion
• Using datetime(d) instead of strptime
• Forgetting to rotate date labels