replace() for value substitution in Pandas - Time & Space Complexity

Time Complexity: replace() for value substitution

O(n x m)

Understanding Time Complexity

We want to understand how the time needed to replace values in a pandas DataFrame changes as the data grows.

How does the replace() method scale when changing many values?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd

df = pd.DataFrame({
    'A': ['cat', 'dog', 'bird', 'cat', 'dog'],
    'B': ['red', 'blue', 'red', 'green', 'blue']
})

# Replace 'cat' with 'lion' and 'blue' with 'cyan'
df_replaced = df.replace({'cat': 'lion', 'blue': 'cyan'})

This code replaces specific values in the DataFrame with new ones.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: pandas scans each cell in the DataFrame to check if it matches any value to replace.
• How many times: Once for each cell, so total cells = number of rows x number of columns.

How Execution Grows With Input

As the DataFrame grows, the number of cells to check grows too.

Input Size (rows x columns)	Approx. Operations
10 x 2 = 20	About 20 checks
100 x 2 = 200	About 200 checks
1000 x 2 = 2000	About 2000 checks

Pattern observation: The operations grow roughly in direct proportion to the number of cells.

Final Time Complexity

Time Complexity: O(n × m)

This means the time to replace values grows linearly with the total number of cells in the DataFrame.

Common Mistake

[X] Wrong: "replace() only checks the columns where replacements are specified, so it runs faster than scanning the whole DataFrame."

[OK] Correct: pandas replace() checks every cell because it does not know where the values appear; it must scan all data to find matches.

Interview Connect

Understanding how replace() scales helps you explain data cleaning steps clearly and shows you know how data size affects performance.

Self-Check

What if we replaced values only in one column instead of the whole DataFrame? How would the time complexity change?

Practice

1. What does the replace() method do in pandas?

easy

A. It sorts the DataFrame based on column values.

B. It deletes rows containing specific values.

C. It adds new columns to the DataFrame.

D. It swaps specified values in a DataFrame or Series with new values.

Solution

1. Step 1: Understand the purpose of replace()

   The replace() method is used to change specific values in pandas data structures.

2. Step 2: Compare with other options

   The other options describe sorting the DataFrame, deleting rows, and adding columns, which are unrelated to value substitution.

3. Final Answer:

   It swaps specified values in a DataFrame or Series with new values. -> Option D

4. Quick Check:

   replace() = value substitution [OK]

Hint: Remember: replace() swaps old values with new ones [OK]

Common Mistakes:

• Confusing replace() with drop() or sort_values()
• Thinking replace() adds or removes rows
• Assuming replace() changes column names

2. Which of the following is the correct syntax to replace all occurrences of 0 with 100 in a pandas Series s?

easy

A. s.replace([100], [0])

B. s.replace(100, 0)

C. s.replace(0, 100)

D. s.replace({100: 0})

Solution

1. Step 1: Identify the correct replace() syntax

   The syntax s.replace(old_value, new_value) replaces all old_value with new_value.

2. Step 2: Check each option

   s.replace(0, 100) correctly replaces 0 with 100. The other options reverse the values: s.replace(100, 0), s.replace([100], [0]), s.replace({100: 0}).

3. Final Answer:

   s.replace(0, 100) -> Option C

4. Quick Check:

   replace(old, new) = s.replace(0, 100) [OK]

Hint: Use s.replace(old_value, new_value) for single value swap [OK]

Common Mistakes:

• Swapping the old and new values by mistake
• Using dict with reversed keys and values
• Confusing list syntax with dict syntax

3. Given the DataFrame:

import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3, 2], 'B': [4, 5, 2, 4]})
df_new = df.replace(2, 99)

What is the content of df_new?

medium

A. {'A': [1, 99, 3, 99], 'B': [4, 5, 99, 4]}

B. {'A': [1, 2, 3, 2], 'B': [4, 5, 2, 4]}

C. {'A': [99, 99, 99, 99], 'B': [99, 99, 99, 99]}

D. {'A': [1, 2, 3, 2], 'B': [4, 5, 99, 4]}

Solution

1. Step 1: Understand replace(2, 99) on DataFrame

   All occurrences of 2 in the DataFrame are replaced by 99.

2. Step 2: Apply replacement to each column

   Column 'A': 2 replaced by 99 at positions 1 and 3.
   Column 'B': 2 replaced by 99 at position 2.

3. Final Answer:

   {'A': [1, 99, 3, 99], 'B': [4, 5, 99, 4]} -> Option A

4. Quick Check:

   All 2s -> 99 in df_new [OK]

Hint: replace() changes all matching values in all columns [OK]

Common Mistakes:

• Assuming replace only affects one column
• Thinking original df changes without assignment
• Replacing only first occurrence instead of all

4. What is wrong with this code snippet?

df = pd.DataFrame({'X': [1, 2, 3]})
df.replace(2, 99)
print(df)

It is intended to replace 2 with 99 in column 'X'.

medium

A. The replace() method does not modify df unless inplace=True or reassigned.

B. The replace() method syntax is incorrect; it needs a dictionary.

C. The print statement should be before replace().

D. replace() cannot be used on DataFrames.

Solution

1. Step 1: Understand replace() behavior

   By default, replace() returns a new DataFrame and does not change the original unless inplace=True or reassigned.

2. Step 2: Analyze the code snippet

   The code calls replace() but does not assign the result back to df or use inplace=True, so df remains unchanged.

3. Final Answer:

   The replace() method does not modify df unless inplace=True or reassigned. -> Option A

4. Quick Check:

   replace() returns new object unless inplace=True [OK]

Hint: Always assign replace() result or use inplace=True [OK]

Common Mistakes:

• Expecting replace() to change original without assignment
• Using wrong syntax for replace()
• Thinking replace() works only on Series

5. You have a DataFrame:

df = pd.DataFrame({'Grade': ['A', 'B', 'C', 'D', 'F']})

You want to replace grades 'A' and 'B' with 'Pass' and 'C', 'D', 'F' with 'Fail' in one call. Which code does this correctly?

hard

A. df.replace(['A', 'B'], ['Pass', 'Pass']).replace(['C', 'D', 'F'], ['Fail', 'Fail', 'Fail'])

B. df.replace({'A': 'Pass', 'B': 'Pass', 'C': 'Fail', 'D': 'Fail', 'F': 'Fail'})

C. df.replace({'Pass': ['A', 'B'], 'Fail': ['C', 'D', 'F']})

D. df.replace({'A': 'Pass', 'B': 'Fail', 'C': 'Pass', 'D': 'Fail', 'F': 'Pass'})

Solution

1. Step 1: Understand replace() with dictionary mapping

   Using a dictionary with keys as old values and values as new values allows multiple replacements in one call.

2. Step 2: Check each option

   df.replace({'A': 'Pass', 'B': 'Pass', 'C': 'Fail', 'D': 'Fail', 'F': 'Fail'}) correctly maps each grade to 'Pass' or 'Fail'. df.replace(['A', 'B'], ['Pass', 'Pass']).replace(['C', 'D', 'F'], ['Fail', 'Fail', 'Fail']) uses two replace calls, which works but is two steps, not one call. df.replace({'Pass': ['A', 'B'], 'Fail': ['C', 'D', 'F']}) reverses keys and values, which is invalid. df.replace({'A': 'Pass', 'B': 'Fail', 'C': 'Pass', 'D': 'Fail', 'F': 'Pass'}) mixes mappings incorrectly.

3. Final Answer:

   df.replace({'A': 'Pass', 'B': 'Pass', 'C': 'Fail', 'D': 'Fail', 'F': 'Fail'}) -> Option B

4. Quick Check:

   Use dict in replace() for multiple value swaps [OK]

Hint: Use dict with old:new pairs for multiple replacements [OK]

Common Mistakes:

• Using lists instead of dict for multiple replacements
• Reversing keys and values in dict
• Chaining replace() unnecessarily