Boolean indexing in Data Analysis Python - Time & Space Complexity

Time Complexity: Boolean indexing

O(n)

Understanding Time Complexity

We want to understand how the time to filter data using Boolean indexing changes as the data size grows.

How does the filtering time increase when we have more data rows?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd

data = pd.DataFrame({
    'age': [23, 45, 12, 36, 52],
    'score': [88, 92, 79, 94, 67]
})

filtered = data[data['age'] > 30]

This code filters rows where the 'age' column is greater than 30.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Checking each row's 'age' value against 30.
• How many times: Once for every row in the data.

How Execution Grows With Input

As the number of rows grows, the filtering checks grow at the same rate.

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

Pattern observation: The number of operations grows directly with the number of rows.

Final Time Complexity

Time Complexity: O(n)

This means the time to filter grows in direct proportion to the number of rows in the data.

Common Mistake

[X] Wrong: "Filtering with Boolean indexing is instant no matter how big the data is."

[OK] Correct: Each row must be checked, so more rows mean more work and more time.

Interview Connect

Understanding how filtering scales helps you explain data processing speed and efficiency clearly in real projects.

Self-Check

"What if we filter using multiple conditions combined with &? How would the time complexity change?"

Practice

1. What does Boolean indexing do in data analysis?

easy

A. Selects data based on true or false conditions

B. Sorts data in ascending order

C. Converts data types automatically

D. Merges two datasets together

Solution

1. Step 1: Understand the purpose of Boolean indexing

   Boolean indexing uses true/false values to filter data.

2. Step 2: Compare with other options

   Sorting, converting types, and merging are different operations.

3. Final Answer:

   Selects data based on true or false conditions -> Option A

4. Quick Check:

   Boolean indexing = filter by true/false [OK]

Hint: Boolean indexing filters data by true/false values [OK]

Common Mistakes:

• Confusing Boolean indexing with sorting
• Thinking it changes data types
• Mixing it up with merging datasets

2. Which of the following is the correct syntax to select rows where column 'age' is greater than 30 in a DataFrame df?

easy

A. df['age' > 30]

B. df[df.age = 30]

C. df[df['age'] > 30]

D. df[df['age'] < 30]

Solution

1. Step 1: Identify correct Boolean indexing syntax

   Use df[condition], where condition is df['age'] > 30.

2. Step 2: Check other options for errors

   df['age' > 30] has wrong brackets, C uses assignment (=) instead of comparison, D uses wrong comparison sign.

3. Final Answer:

   df[df['age'] > 30] -> Option C

4. Quick Check:

   Correct syntax uses df[condition] with comparison [OK]

Hint: Use df[condition] with comparison operators for filtering [OK]

Common Mistakes:

• Using assignment '=' instead of comparison '>'
• Wrong bracket placement
• Using incorrect comparison signs

3. Given the DataFrame df below:

import pandas as pd
data = {'score': [85, 42, 78, 90, 55]}
df = pd.DataFrame(data)

What is the output of df[df['score'] >= 78]?

medium

A. Rows with scores 85, 78, and 90

B. Rows with scores 42 and 55

C. All rows because condition is ignored

D. Empty DataFrame

Solution

1. Step 1: Apply the condition df['score'] >= 78

   This selects rows where score is 78 or more: 85, 78, 90.

2. Step 2: Check which rows match

   Rows with scores 42 and 55 do not match, so they are excluded.

3. Final Answer:

   Rows with scores 85, 78, and 90 -> Option A

4. Quick Check:

   Filter keeps scores >= 78 [OK]

Hint: Filter keeps rows where condition is true [OK]

Common Mistakes:

• Including rows that don't meet the condition
• Thinking condition is ignored
• Expecting empty output incorrectly

4. What is wrong with this code snippet?

import pandas as pd
data = {'value': [10, 20, 30, 40]}
df = pd.DataFrame(data)
filtered = df[df.value > 25]

medium

A. It will raise a syntax error

B. It will raise an AttributeError

C. It returns an empty DataFrame

D. It works correctly and filters rows with value > 25

Solution

1. Step 1: Check attribute access for column 'value'

   Using df.value is valid if column name has no spaces or special chars.

2. Step 2: Verify filtering condition

   df.value > 25 selects rows with values 30 and 40 correctly.

3. Final Answer:

   It works correctly and filters rows with value > 25 -> Option D

4. Quick Check:

   Attribute access works for simple column names [OK]

Hint: Use df.column_name if no spaces or special chars [OK]

Common Mistakes:

• Assuming df.value always causes error
• Expecting syntax error
• Thinking it returns empty DataFrame

5. You have a DataFrame df with a column 'status' containing values 'pass', 'fail', and 'pending'. How do you select only rows where 'status' is either 'pass' or 'pending' using Boolean indexing?

hard

A. df[df['status'] == 'pass' and 'pending']

B. df[(df['status'] == 'pass') | (df['status'] == 'pending')]

C. df[df['status'] in ['pass', 'pending']]

D. df[df['status'] == 'pass' or df['status'] == 'pending']

Solution

1. Step 1: Use element-wise OR for multiple conditions

   Use | between conditions inside parentheses for correct Boolean indexing.

2. Step 2: Avoid Python keywords and wrong syntax

   'and' and 'or' do not work element-wise; 'in' is not vectorized here.

3. Final Answer:

   df[(df['status'] == 'pass') | (df['status'] == 'pending')] -> Option B

4. Quick Check:

   Use | with parentheses for multiple conditions [OK]

Hint: Use | with parentheses for multiple conditions [OK]

Common Mistakes:

• Using 'and' or 'or' instead of | and &
• Trying to use 'in' directly on Series
• Missing parentheses around conditions