Why boolean masking matters in NumPy - Performance Analysis

Time Complexity: Why boolean masking matters

O(n)

Understanding Time Complexity

We want to see how fast numpy handles selecting data using boolean masks.

How does the time to pick items grow when the data gets bigger?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import numpy as np

arr = np.arange(1000000)
mask = arr % 2 == 0
filtered = arr[mask]

This code creates a large array, makes a mask for even numbers, and selects those numbers.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Checking each element to see if it is even (creating the mask).
• How many times: Once for every element in the array.
• Secondary operation: Using the mask to pick elements (also touches each element once).

How Execution Grows With Input

As the array gets bigger, the time to check and select grows in a straight line.

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

Pattern observation: The work grows directly with the number of items.

Final Time Complexity

Time Complexity: O(n)

This means the time to filter grows in a straight line as the data size grows.

Common Mistake

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

[OK] Correct: The mask must check every item, so bigger data means more work and more time.

Interview Connect

Understanding how boolean masking scales helps you explain data filtering clearly and confidently in real tasks.

Self-Check

"What if we used multiple conditions combined in the mask? How would the time complexity change?"

Practice

1. What is the main purpose of boolean masking in numpy?

easy

A. To sort an array in ascending order

B. To select elements from an array based on True/False conditions

C. To change the data type of an array

D. To create a new array filled with zeros

Solution

1. Step 1: Understand boolean masking concept

   Boolean masking uses a True/False array to pick elements from another array.

2. Step 2: Identify the main use

   This helps select only the elements where the mask is True, filtering data easily.

3. Final Answer:

   To select elements from an array based on True/False conditions -> Option B

4. Quick Check:

   Boolean mask = select elements [OK]

Hint: Boolean mask picks elements where condition is True [OK]

Common Mistakes:

• Thinking it sorts the array
• Confusing masking with data type change
• Assuming it fills arrays with zeros

2. Which of the following is the correct syntax to create a boolean mask for array arr to select values greater than 5?

easy

A. mask = arr > 5

B. mask = arr = 5

C. mask = arr < 5

D. mask = arr == 5

Solution

1. Step 1: Understand comparison operators

   To select values greater than 5, use the greater than operator: >.

2. Step 2: Check syntax correctness

   mask = arr > 5 creates a boolean array where True means element > 5.

3. Final Answer:

   mask = arr > 5 -> Option A

4. Quick Check:

   Use > for greater than [OK]

Hint: Use > operator to create mask for values greater than number [OK]

Common Mistakes:

• Using single equals (=) instead of comparison (>)
• Using < instead of >
• Using == which checks equality, not greater than

3. Given the code:

import numpy as np
arr = np.array([2, 7, 4, 9, 1])
mask = arr > 4
result = arr[mask]

What is the value of result?

medium

A. [2, 7, 9]

B. [2, 4, 1]

C. [7, 4, 9]

D. [7, 9]

Solution

1. Step 1: Create boolean mask for elements > 4

   Elements greater than 4 are 7 and 9, so mask is [False, True, False, True, False].

2. Step 2: Apply mask to array

   Using arr[mask] selects elements where mask is True: [7, 9].

3. Final Answer:

   [7, 9] -> Option D

4. Quick Check:

   Mask picks elements > 4 [OK]

Hint: Mask True picks elements, False skips [OK]

Common Mistakes:

• Including elements not > 4
• Confusing mask with index positions
• Selecting elements less than or equal to 4

4. What is wrong with this code snippet?

import numpy as np
arr = np.array([1, 3, 5, 7])
mask = arr > 4
print(arr[mask])

It raises an error. Why?

medium

A. The mask is created correctly; no error occurs

B. The mask uses assignment (=) instead of comparison (>)

C. The array contains non-numeric values causing error

D. The mask array has different length than arr

Solution

1. Step 1: Check mask creation

   The mask arr > 4 creates a boolean array of same length as arr without error.

2. Step 2: Check indexing with mask

   Using arr[mask] selects elements > 4 without error.

3. Final Answer:

   The mask is created correctly; no error occurs -> Option A

4. Quick Check:

   Correct mask syntax means no error [OK]

Hint: Correct mask syntax means no error [OK]

Common Mistakes:

• Confusing assignment (=) with comparison (>)
• Assuming mask length mismatch error
• Thinking non-numeric values cause error here

5. You have a numpy array data = np.array([10, 0, 5, -3, 8]). You want to select only positive numbers excluding zero using boolean masking. Which code correctly achieves this?

hard

A. mask = data >= 0
result = data[mask]

B. mask = data != 0
result = data[mask]

C. mask = data > 0
result = data[mask]

D. mask = data < 0
result = data[mask]

Solution

1. Step 1: Define condition for positive numbers excluding zero

   Positive numbers are greater than zero, so condition is data > 0.

2. Step 2: Apply mask and select elements

   Using data[data > 0] selects 10, 5, and 8, excluding zero and negatives.

3. Final Answer:

   mask = data > 0 result = data[mask] -> Option C

4. Quick Check:

   Use > 0 to exclude zero and negatives [OK]

Hint: Use > 0 to select positive numbers excluding zero [OK]

Common Mistakes:

• Using >= 0 includes zero
• Using != 0 includes negatives
• Using < 0 selects negatives, not positives