Removing duplicates (drop_duplicates) in Data Analysis Python - Time & Space Complexity

Time Complexity: Removing duplicates (drop_duplicates)

O(n)

Understanding Time Complexity

When we remove duplicates from data, we want to know how the time to do this changes as the data grows.

We ask: How much longer does it take if we have more rows?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd

data = pd.DataFrame({
    'A': [1, 2, 2, 3, 4, 4, 4],
    'B': ['x', 'y', 'y', 'z', 'x', 'x', 'x']
})

unique_data = data.drop_duplicates()

This code creates a table and removes rows that appear more than once.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Checking each row against others to find duplicates.
• How many times: Each row is hashed or checked once during the process.

How Execution Grows With Input

As the number of rows grows, the work to find duplicates grows roughly in a straight line.

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

Pattern observation: Doubling the rows roughly doubles the work.

Final Time Complexity

Time Complexity: O(n)

This means the time to remove duplicates grows directly with the number of rows.

Common Mistake

[X] Wrong: "Removing duplicates takes much longer than just reading the data because it compares every row to every other row."

[OK] Correct: The method uses smart hashing or indexing to avoid checking every pair, so it does not compare all rows to all others.

Interview Connect

Understanding how removing duplicates scales helps you explain data cleaning steps clearly and shows you know how data size affects processing time.

Self-Check

"What if we remove duplicates based on only one column instead of all columns? How would the time complexity change?"

Practice

1. What does the drop_duplicates() function do in a pandas DataFrame?

easy

A. Fills missing values with zeros

B. Removes repeated rows based on all columns by default

C. Sorts the DataFrame rows in ascending order

D. Changes the data type of columns

Solution

1. Step 1: Understand the function purpose

   drop_duplicates() is used to remove repeated rows in a DataFrame.

2. Step 2: Default behavior

   By default, it checks all columns to find duplicates and keeps the first occurrence.

3. Final Answer:

   Removes repeated rows based on all columns by default -> Option B

4. Quick Check:

   drop_duplicates() removes repeated rows [OK]

Hint: Remember: drop_duplicates removes repeated rows [OK]

Common Mistakes:

• Thinking it sorts data
• Confusing with fillna()
• Assuming it changes data types

2. Which of the following is the correct syntax to remove duplicate rows from a DataFrame df?

easy

A. df.drop_duplicates[]

B. df.remove_duplicates()

C. df.drop_duplicates()

D. df.drop_duplicates{}

Solution

1. Step 1: Identify correct method call

   The method to remove duplicates is drop_duplicates(), called with parentheses.

2. Step 2: Check syntax correctness

   Only df.drop_duplicates() is valid syntax; brackets or braces are incorrect.

3. Final Answer:

   df.drop_duplicates() -> Option C

4. Quick Check:

   Use parentheses to call methods [OK]

Hint: Use parentheses () to call drop_duplicates method [OK]

Common Mistakes:

• Using square brackets instead of parentheses
• Using a non-existent method name
• Using curly braces instead of parentheses

3. Given the DataFrame df below, what will df.drop_duplicates(subset=['Name']) return?

import pandas as pd
data = {'Name': ['Anna', 'Bob', 'Anna', 'Cody'], 'Age': [25, 30, 25, 22]}
df = pd.DataFrame(data)

medium

A. [{'Name': 'Anna', 'Age': 25}, {'Name': 'Bob', 'Age': 30}, {'Name': 'Cody', 'Age': 22}]

B. [{'Name': 'Anna', 'Age': 25}, {'Name': 'Anna', 'Age': 25}, {'Name': 'Bob', 'Age': 30}, {'Name': 'Cody', 'Age': 22}]

C. [{'Name': 'Bob', 'Age': 30}, {'Name': 'Cody', 'Age': 22}]

D. [{'Name': 'Anna', 'Age': 25}, {'Name': 'Bob', 'Age': 30}]

Solution

1. Step 1: Understand subset parameter

   Using subset=['Name'] means duplicates are checked only on the 'Name' column.

2. Step 2: Identify duplicates by 'Name'

   'Anna' appears twice; only the first occurrence is kept, others dropped.

3. Final Answer:

   [{'Name': 'Anna', 'Age': 25}, {'Name': 'Bob', 'Age': 30}, {'Name': 'Cody', 'Age': 22}] -> Option A

4. Quick Check:

   Duplicates dropped by 'Name' column [OK]

Hint: Use subset to check duplicates on specific columns [OK]

Common Mistakes:

• Not using subset and dropping all duplicates
• Keeping all duplicates instead of first
• Confusing which rows are dropped

4. What is wrong with this code snippet to remove duplicates from DataFrame df?

df.drop_duplicates(inplace=True)

After running it, duplicates still appear in df. Why?

medium

A. The DataFrame has no duplicates to remove

B. drop_duplicates() does not modify the original DataFrame even with inplace=True

C. The DataFrame was not reassigned and inplace=True is ignored

D. The inplace parameter is misspelled

Solution

1. Step 1: Check inplace parameter usage

   inplace=True modifies the original DataFrame, so no reassignment needed.

2. Step 2: Consider DataFrame content

   If duplicates still appear, likely the DataFrame had no duplicates to remove.

3. Final Answer:

   The DataFrame has no duplicates to remove -> Option A

4. Quick Check:

   inplace=True modifies original; no duplicates means no change [OK]

Hint: Check if DataFrame actually has duplicates before drop_duplicates [OK]

Common Mistakes:

• Assuming inplace=True does nothing
• Thinking drop_duplicates always changes data
• Not verifying if duplicates exist

5. You have a DataFrame df with columns ['ID', 'Name', 'Score']. Some rows have duplicate ID values but different Score. How do you keep only the row with the highest Score for each ID?

hard

A. Use df.drop_duplicates(subset=['Score'])

B. Use df.drop_duplicates(subset=['ID']) without sorting

C. Use df.drop_duplicates(keep='last') without sorting

D. Use df.sort_values('Score', ascending=False).drop_duplicates(subset=['ID'])

Solution

1. Step 1: Sort DataFrame by Score descending

   Sorting by 'Score' descending puts highest scores first for each ID.

2. Step 2: Drop duplicates by ID keeping first

   After sorting, dropping duplicates by 'ID' keeps the first (highest score) row per ID.

3. Final Answer:

   df.sort_values('Score', ascending=False).drop_duplicates(subset=['ID']) -> Option D

4. Quick Check:

   Sort then drop duplicates keeps highest score per ID [OK]

Hint: Sort by score descending before drop_duplicates to keep max [OK]

Common Mistakes:

• Not sorting before dropping duplicates
• Dropping duplicates on wrong column
• Using keep='last' without sorting