Regex operations in Pandas - Time & Space Complexity

Time Complexity: Regex operations in Pandas

O(n)

Understanding Time Complexity

When we use regex operations in pandas, we want to know how the time it takes changes as our data grows.

We ask: How does searching or matching patterns slow down when we have more data?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd

df = pd.DataFrame({
    'text': ['apple123', 'banana456', 'cherry789', 'date012'] * 1000
})

matches = df['text'].str.contains(r'\d{3}')

This code checks each string in the 'text' column to see if it contains three digits in a row.

Identify Repeating Operations

• Primary operation: Applying the regex pattern to each string in the column.
• How many times: Once for every row in the DataFrame.

How Execution Grows With Input

As the number of rows grows, the total work grows roughly in the same way.

Input Size (n)	Approx. Operations
10	About 10 regex checks
100	About 100 regex checks
1000	About 1000 regex checks

Pattern observation: Doubling the rows roughly doubles the work because each row is checked once.

Final Time Complexity

Time Complexity: O(n)

This means the time grows linearly with the number of rows; more rows mean proportionally more work.

Common Mistake

[X] Wrong: "Regex operations take constant time no matter how many rows there are."

[OK] Correct: Each row is checked separately, so more rows mean more checks and more time.

Interview Connect

Understanding how regex operations scale helps you explain your code's speed and handle bigger data confidently.

Self-Check

"What if we changed the regex to a more complex pattern that takes longer to match? How would the time complexity change?"

Practice

1. What does the str.contains method do in Pandas when used with a regex pattern?

easy

A. It extracts the first match of the regex pattern from each string.

B. It replaces all matches of the regex pattern with a new string.

C. It sorts the DataFrame rows based on regex matches.

D. It checks if each string in a column contains the regex pattern and returns true or false.

Solution

1. Step 1: Understand str.contains usage

   str.contains checks if the regex pattern appears anywhere in each string of the column and returns a boolean Series.

2. Step 2: Differentiate from other methods

   str.replace changes text, str.extract pulls out matches, and sorting is unrelated.

3. Final Answer:

   It checks if each string in a column contains the regex pattern and returns true or false. -> Option D

4. Quick Check:

   str.contains = Boolean check [OK]

Hint: Contains returns true/false if pattern found in each string [OK]

Common Mistakes:

• Confusing contains with replace or extract
• Thinking contains modifies the data
• Assuming contains returns matched text

2. Which of the following is the correct syntax to replace every sequence of consecutive digits in a Pandas column df['col'] with the character 'X' using regex?

easy

A. df['col'].str.replace('\d+', 'X')

B. df['col'].str.replace('[0-9]', 'X')

C. df['col'].replace('[0-9]', 'X')

D. df['col'].str.extract('[0-9]', 'X')

Solution

1. Step 1: Identify correct method for regex replacement

   Use str.replace on the Series to replace regex matches. df['col'].replace('[0-9]', 'X') is wrong because replace without str is for exact values, not regex.

2. Step 2: Check regex pattern and syntax

   df['col'].str.replace('[0-9]', 'X') matches single digits only ('123' -> 'XXX'). df['col'].str.replace('\d+', 'X') matches sequences of one or more digits ('123' -> 'X').

3. Final Answer:

   df['col'].str.replace('\d+', 'X') -> Option A

4. Quick Check:

   Use str.replace with regex pattern [OK]

Hint: Use str.replace with regex pattern and replacement string [OK]

Common Mistakes:

• Using DataFrame replace instead of Series.str.replace
• Wrong regex pattern syntax
• Using str.extract instead of replace

3. Given the DataFrame:

import pandas as pd
df = pd.DataFrame({'text': ['apple123', 'banana', 'cherry456', 'date789']})
df['numbers'] = df['text'].str.extract('(\d+)')

What will be the content of the new column numbers?

medium

A. ['123', 'banana', '456', '789']

B. ['apple', NaN, 'cherry', 'date']

C. ['123', NaN, '456', '789']

D. [NaN, NaN, NaN, NaN]

Solution

1. Step 1: Understand str.extract with regex

   The regex (\d+) captures one or more digits. str.extract returns the first match or NaN if none.

2. Step 2: Apply regex to each string

   'apple123' matches '123', 'banana' has no digits so NaN, 'cherry456' matches '456', 'date789' matches '789'.

3. Final Answer:

   ['123', NaN, '456', '789'] -> Option C

4. Quick Check:

   Extract digits or NaN if none [OK]

Hint: Extract returns first regex group or NaN if no match [OK]

Common Mistakes:

• Expecting extract to return full string
• Confusing extract with contains or replace
• Not handling NaN for no matches

4. Consider this code snippet:

import pandas as pd
df = pd.DataFrame({'col': ['abc123', 'def456', 'ghi789']})
df['col'].str.match('[a-z]+\d')

What is the error or issue in this code?

medium

A. The regex pattern misses the last digit, so match returns false for all rows.

B. The code raises a SyntaxError due to incorrect regex syntax.

C. The method str.match does not accept regex patterns.

D. The code runs correctly and returns true for all rows.

Solution

1. Step 1: Analyze the regex pattern

   The pattern [a-z]+\d matches one or more lowercase letters followed by exactly one digit from the start of the string.

2. Step 2: Compare pattern with data strings

   The strings 'abc123', 'def456', 'ghi789' have multiple digits at the end, but the pattern only expects one digit. Since str.match matches from the start and the pattern does not cover the full digit sequence, it returns False for all rows.

3. Final Answer:

   The regex pattern misses the last digit, so match returns false for all rows. -> Option A

4. Quick Check:

   str.match matches from start; pattern must cover full prefix for True [OK]

Hint: str.match anchors at start (prefix match), pattern must cover full prefix [OK]

Common Mistakes:

• Thinking str.match requires full string match
• Confusing str.match with str.fullmatch
• Thinking regex syntax is invalid here

5. You have a DataFrame df with a column 'emails' containing email addresses mixed with invalid entries. How would you create a new column 'valid' that is true only for rows where 'emails' contains a valid email pattern using regex in Pandas?

hard

A. df['valid'] = df['emails'].str.extract(r'^[\w\.-]+@[\w\.-]+\.\w+$')

B. df['valid'] = df['emails'].str.contains(r'^[\w\.-]+@[\w\.-]+\.\w+$')

C. df['valid'] = df['emails'].str.replace(r'^[\w\.-]+@[\w\.-]+\.\w+$', 'true')

D. df['valid'] = df['emails'].str.match(r'\w+@\w+')

Solution

1. Step 1: Choose method to check pattern presence

   str.contains returns true/false if regex matches anywhere in string, suitable for validation.

2. Step 2: Use correct regex for full email validation

   The regex ^[\w\.-]+@[\w\.-]+\.\w+$ matches a full email pattern from start to end. df['valid'] = df['emails'].str.extract(r'^[\w\.-]+@[\w\.-]+\.\w+$') extracts matched text, not boolean. df['valid'] = df['emails'].str.replace(r'^[\w\.-]+@[\w\.-]+\.\w+$', 'true') replaces text, not boolean. df['valid'] = df['emails'].str.match(r'\w+@\w+') uses str.match but regex is incomplete and may match partial emails.

3. Final Answer:

   df['valid'] = df['emails'].str.contains(r'^[\w\.-]+@[\w\.-]+\.\w+$') -> Option B

4. Quick Check:

   Use str.contains with full regex for validation [OK]

Hint: Use str.contains with full regex pattern for true/false validation [OK]

Common Mistakes:

• Using str.extract which returns strings, not booleans
• Using str.replace which changes data instead of checking
• Using incomplete regex that matches partial emails