Date arithmetic with timedelta in Pandas - Time & Space Complexity

Time Complexity: Date arithmetic with timedelta

O(n)

Understanding Time Complexity

We want to understand how the time it takes to do date calculations changes as we work with more dates.

Specifically, how does adding or subtracting time from many dates affect the work done?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd
from datetime import timedelta

dates = pd.date_range('2023-01-01', periods=1000)
delta = timedelta(days=5)

new_dates = dates + delta

This code creates 1000 dates and adds 5 days to each date.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Adding a fixed time difference to each date in the list.
• How many times: Once for each date, so 1000 times in this example.

How Execution Grows With Input

When we add a time difference to each date, the work grows as we have more dates.

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

Pattern observation: The work grows directly with the number of dates. Double the dates, double the work.

Final Time Complexity

Time Complexity: O(n)

This means the time to add a timedelta grows in a straight line with the number of dates.

Common Mistake

[X] Wrong: "Adding a timedelta to a list of dates happens instantly regardless of list size."

[OK] Correct: Each date must be updated one by one, so more dates mean more work and more time.

Interview Connect

Understanding how operations scale with data size helps you write efficient code and explain your choices clearly.

Self-Check

"What if we added a different timedelta to each date instead of the same one? How would the time complexity change?"

Practice

1. What does pd.Timedelta(days=3) represent in pandas?

easy

A. A string describing 3 days

B. A specific date 3 days from now

C. A time difference of 3 days

D. A timestamp at 3 AM

Solution

1. Step 1: Understand Timedelta purpose

   pd.Timedelta is used to represent a duration or difference in time, not a fixed date or time.

2. Step 2: Interpret the argument

   The argument days=3 means the duration is 3 days long.

3. Final Answer:

   A time difference of 3 days -> Option C

4. Quick Check:

   Timedelta(days=3) = 3 days [OK]

Hint: Timedelta means duration, not a fixed date [OK]

Common Mistakes:

• Thinking Timedelta is a fixed date
• Confusing Timedelta with Timestamp
• Assuming it returns a string

2. Which of the following is the correct way to add 5 days to a pandas Timestamp ts?

easy

A. ts.add(pd.Timedelta(5))

B. ts + pd.Timedelta('5')

C. ts + pd.Timedelta(5 days)

D. ts + pd.Timedelta(days=5)

Solution

1. Step 1: Check correct Timedelta syntax

   The correct way to specify 5 days is pd.Timedelta(days=5). The argument must be named or a valid string.

2. Step 2: Verify addition syntax

   Adding a Timedelta to a Timestamp uses the + operator, so ts + pd.Timedelta(days=5) is correct.

3. Final Answer:

   ts + pd.Timedelta(days=5) -> Option D

4. Quick Check:

   Correct syntax uses days=5 inside Timedelta [OK]

Hint: Use named argument days=5 inside Timedelta [OK]

Common Mistakes:

• Missing quotes or wrong argument format
• Using add() method incorrectly
• Passing string without units

3. What is the output of this code?

import pandas as pd
start = pd.Timestamp('2024-01-01')
new_date = start + pd.Timedelta(days=10) - pd.Timedelta(days=3)
print(new_date.strftime('%Y-%m-%d'))

medium

A. 2024-01-08

B. 2024-01-13

C. 2024-01-04

D. 2024-01-10

Solution

1. Step 1: Calculate addition of 10 days

   Starting from 2024-01-01, adding 10 days gives 2024-01-11.

2. Step 2: Subtract 3 days from the result

   Subtracting 3 days from 2024-01-11 results in 2024-01-08.

3. Final Answer:

   2024-01-08 -> Option A

4. Quick Check:

   2024-01-01 + 10 - 3 days = 2024-01-08 [OK]

Hint: Add then subtract days stepwise to find final date [OK]

Common Mistakes:

• Adding and subtracting days in wrong order
• Miscounting days
• Confusing string formatting output

4. The following code throws an error. What is the problem?

import pandas as pd
start = pd.Timestamp('2024-05-01')
new_date = start + pd.Timedelta(5)
print(new_date)

medium

A. Timedelta needs a unit like days=5

B. Timestamp cannot be added to Timedelta

C. pd.Timedelta requires string input only

D. Missing import for Timedelta

Solution

1. Step 1: Check Timedelta argument

   pd.Timedelta(5) is missing the unit. It needs a keyword like days=5 or a string like '5 days'.

2. Step 2: Understand error cause

   Without specifying units, pandas cannot interpret the number 5 alone, causing an error.

3. Final Answer:

   Timedelta needs a unit like days=5 -> Option A

4. Quick Check:

   Timedelta(5) missing units causes error [OK]

Hint: Always specify units when creating Timedelta [OK]

Common Mistakes:

• Passing integer without units
• Assuming default unit is days
• Confusing Timestamp and Timedelta usage

5. You have a pandas DataFrame df with a column 'date' of type datetime. You want to create a new column 'date_plus_7' that adds 7 days to each date. Which code correctly does this?

hard

A. df['date_plus_7'] = df['date'] + pd.Timedelta('7 days')

B. df['date_plus_7'] = df['date'] + pd.Timedelta(days=7)

C. df['date_plus_7'] = df['date'] + pd.Timedelta(7)

D. df['date_plus_7'] = df['date'] + pd.Timedelta(days=7) * len(df)

Solution

1. Step 1: Understand Timedelta broadcasting

   Adding a Timedelta to a pandas Series of dates works if the Timedelta is a scalar (single duration).

2. Step 2: Check each option

   df['date_plus_7'] = df['date'] + pd.Timedelta('7 days') uses a string '7 days' which works but is less common in vectorized operations. df['date_plus_7'] = df['date'] + pd.Timedelta(days=7) uses pd.Timedelta(days=7) which is the standard and recommended way. df['date_plus_7'] = df['date'] + pd.Timedelta(7) misses units and causes error. df['date_plus_7'] = df['date'] + pd.Timedelta(days=7) * len(df) multiplies timedelta by length, which incorrectly scales the timedelta.

3. Step 3: Choose the best practice

   df['date_plus_7'] = df['date'] + pd.Timedelta(days=7) is the correct and recommended way to add 7 days to each date in a column.

4. Final Answer:

   df['date_plus_7'] = df['date'] + pd.Timedelta(days=7) -> Option B

5. Quick Check:

   Use pd.Timedelta(days=7) for vectorized addition with Series [OK]

Hint: Use pd.Timedelta(days=7) for vectorized addition with Series [OK]

Common Mistakes:

• Passing integer without units
• Multiplying Timedelta by length of DataFrame
• Not using vectorized Timedelta correctly