Missing data strategies decision in Pandas - Time & Space Complexity

Time Complexity: Missing data strategies decision

O(n)

Understanding Time Complexity

When working with missing data in pandas, it is important to know how the time to handle missing values changes as the data grows.

We want to understand how the cost of different missing data strategies grows with the size of the dataset.

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd

df = pd.DataFrame({
    'A': [1, 2, None, 4, None],
    'B': [None, 2, 3, None, 5]
})

# Strategy: Drop rows with any missing values
df_clean = df.dropna()

# Strategy: Fill missing values with a constant
df_filled = df.fillna(0)

This code shows two common ways to handle missing data: dropping rows with missing values and filling missing values with a constant.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Scanning each row and column to check for missing values.
• How many times: Once for each cell in the DataFrame (rows x columns).

How Execution Grows With Input

As the number of rows grows, the operations to find and handle missing values increase proportionally.

Input Size (rows)	Approx. Operations
10	About 10 x columns checks
100	About 100 x columns checks
1000	About 1000 x columns checks

Pattern observation: The work grows linearly with the number of rows.

Final Time Complexity

Time Complexity: O(n)

This means the time to handle missing data grows roughly in direct proportion to the number of rows in the data.

Common Mistake

[X] Wrong: "Handling missing data takes the same time no matter how big the dataset is."

[OK] Correct: The code must check each cell for missing values, so more data means more checks and more time.

Interview Connect

Understanding how missing data handling scales helps you explain your data cleaning choices clearly and confidently in real projects.

Self-Check

"What if we used a method that fills missing values based on the mean of each column? How would the time complexity change?"

Practice

1.

What is the main purpose of using dropna() in pandas?

easy

A. To remove rows or columns that contain missing data

B. To replace missing data with zeros

C. To fill missing data with the mean value

D. To convert data types of columns

Solution

1. Step 1: Understand dropna() function

   dropna() removes rows or columns that have missing values (NaN).

2. Step 2: Compare with other options

   Replacing missing data or converting data types are done by other functions like fillna() or astype().

3. Final Answer:

   To remove rows or columns that contain missing data -> Option A

4. Quick Check:

   dropna() removes missing data rows/columns [OK]

Hint: Remember: dropna drops missing data, fillna fills it [OK]

Common Mistakes:

• Confusing dropna with fillna
• Thinking dropna fills missing values
• Assuming dropna changes data types

2.

Which of the following is the correct syntax to fill missing values in a DataFrame df with the value 0?

?

easy

A. df.fillna(0)

B. df.dropna(0)

C. df.fill(na=0)

D. df.replace(na=0)

Solution

1. Step 1: Recall fillna syntax

   The correct method to fill missing values is fillna() with the value as argument.

2. Step 2: Check each option

   df.fillna(0) uses df.fillna(0) which is correct. Others have wrong method names or parameters.

3. Final Answer:

   df.fillna(0) -> Option A

4. Quick Check:

   Use fillna(value) to fill missing values [OK]

Hint: fillna fills missing values; dropna removes rows/columns [OK]

Common Mistakes:

• Using dropna instead of fillna
• Wrong method name like fill() or replace()
• Passing parameters incorrectly

3.

Given the DataFrame df below, what will be the output of df.fillna(method='ffill')?

import pandas as pd
df = pd.DataFrame({'A': [1, None, 3, None], 'B': [None, 2, None, 4]})
print(df.fillna(method='ffill'))

medium

A. Removes rows with missing values

B. Fills missing values with zero

C. Fills missing values by propagating last valid observation forward

D. Raises an error due to wrong method

Solution

1. Step 1: Understand forward fill method

   method='ffill' fills missing values by copying the last known non-missing value forward.

2. Step 2: Apply to DataFrame

   For column A: 1 stays, None becomes 1, 3 stays, None becomes 3. For B: None stays None (no previous), 2 stays, None becomes 2, 4 stays.

3. Final Answer:

   Fills missing values by propagating last valid observation forward -> Option C

4. Quick Check:

   ffill copies last valid value forward [OK]

Hint: ffill copies previous value forward to fill NaNs [OK]

Common Mistakes:

• Thinking fillna(method='ffill') fills with zero
• Confusing ffill with dropping rows
• Expecting error from method parameter

4.

What is wrong with the following code if the goal is to remove all rows with missing data from DataFrame df?

df.fillna(drop=True)

medium

A. The code should use df.dropna(inplace=True) to modify in place

B. fillna() does not have a drop parameter; use dropna() instead

C. The parameter drop=True should be dropna=True

D. The code is correct and will remove rows with missing data

Solution

1. Step 1: Check method and parameters

   fillna() is for filling missing values and does not accept a drop parameter.

2. Step 2: Correct method for removing rows

   To remove rows with missing data, use dropna() instead.

3. Final Answer:

   fillna() does not have a drop parameter; use dropna() instead -> Option B

4. Quick Check:

   Use dropna() to remove missing data rows [OK]

Hint: Use dropna() to remove rows, fillna() to replace values [OK]

Common Mistakes:

• Using fillna() with invalid parameters
• Confusing fillna and dropna functions
• Expecting fillna to remove rows

5.

You have a DataFrame df with many missing values scattered randomly. You want to keep as much data as possible but avoid rows with more than 2 missing values. Which pandas method and parameter combination is best?

hard

A. Use df.fillna(method='bfill') to fill missing values backward

B. Use df.fillna(0) to replace all missing values with zero

C. Use df.dropna(axis=1) to drop columns with missing values

D. Use df.dropna(thresh=len(df.columns)-2) to drop rows with more than 2 missing values

Solution

1. Step 1: Understand the threshold parameter

   dropna(thresh=n) keeps rows with at least n non-missing values.

2. Step 2: Calculate threshold for max 2 missing values

   If total columns are len(df.columns), then rows must have at least len(df.columns) - 2 non-missing values to keep.

3. Step 3: Evaluate other options

   Filling with zero or backward fill changes data but does not drop rows. Dropping columns removes data unnecessarily.

4. Final Answer:

   Use df.dropna(thresh=len(df.columns)-2) to drop rows with more than 2 missing values -> Option D

5. Quick Check:

   dropna(thresh) keeps rows with enough non-NaNs [OK]

Hint: Use dropna(thresh) to keep rows with few missing values [OK]

Common Mistakes:

• Using fillna instead of dropping rows
• Dropping columns instead of rows
• Misunderstanding thresh parameter