Creating boolean arrays in NumPy - Visual Walkthrough

Concept Flow - Creating boolean arrays

Start

↓

Choose size or shape

↓

Use numpy boolean array function

↓

Array created with True/False values

↓

Use array for filtering or conditions

↓

End

This flow shows how to create a boolean array in numpy by choosing size, using a function, and then using the array.

Execution Sample

NumPy

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

Create a boolean array by comparing elements of a numpy array to 3.

Execution Table

Step	Action	Expression	Result	Output
1	Create numpy array	np.array([1, 2, 3, 4, 5])	[1 2 3 4 5]	[1 2 3 4 5]
2	Compare array elements > 3	arr > 3	[False False False True True]	[False False False True True]
3	Print boolean array	print(bool_arr)	None	[False False False True True]
4	End	-	-	-

💡 All steps completed, boolean array created and printed.

Variable Tracker

Variable	Start	After Step 1	After Step 2	Final
arr	undefined	[1 2 3 4 5]	[1 2 3 4 5]	[1 2 3 4 5]
bool_arr	undefined	undefined	[False False False True True]	[False False False True True]

Key Moments - 3 Insights

Why does 'arr > 3' create a boolean array?

Because 'arr > 3' compares each element to 3 and returns True or False for each, as shown in execution_table step 2.

Is the original array changed when creating the boolean array?

No, the original array 'arr' stays the same, only 'bool_arr' holds True/False values, as seen in variable_tracker.

What type of values does a boolean array hold?

It holds only True or False values, representing conditions or filters, as shown in the output column of execution_table.

Visual Quiz - 3 Questions

Look at the execution table, what is the value of 'bool_arr' after step 2?

A [False False False True True]

B [True True True False False]

C [1 2 3 4 5]

D [True False True False True]

💡 Hint

Check the 'Result' column in execution_table row for step 2.

At which step is the original array 'arr' created?

A Step 3

B Step 2

C Step 1

D Step 4

💡 Hint

Look at the 'Action' column in execution_table for when 'arr' is assigned.

If we change the condition to 'arr < 3', what will 'bool_arr' contain?

A [False False False True True]

B [True True False False False]

C [False False True True True]

D [True False True False True]

💡 Hint

Think about which elements in [1 2 3 4 5] are less than 3.

Concept Snapshot

Creating boolean arrays in numpy:
- Use comparison operators on numpy arrays (e.g., arr > 3)
- Result is a boolean array of True/False
- Original array stays unchanged
- Boolean arrays help filter or select data
- Use np.array() to create arrays first

Full Transcript

This lesson shows how to create boolean arrays in numpy by comparing array elements to a value. First, we create a numpy array 'arr' with numbers. Then, we compare each element to 3 using 'arr > 3'. This produces a boolean array 'bool_arr' with True where the condition is met and False otherwise. The original array remains unchanged. Boolean arrays are useful for filtering data or conditions in numpy.

Practice

1. What does a boolean array in numpy represent?

easy

A. An array of integers only

B. An array of True/False values based on a condition

C. An array of strings

D. An array with only zeros

Solution

1. Step 1: Understand boolean arrays

   A boolean array stores True or False for each element, usually from a condition.

2. Step 2: Identify the correct description

   The description 'An array of True/False values based on a condition' matches what a boolean array represents. Others refer to integers, strings, or zeros.

3. Final Answer:

   An array of True/False values based on a condition -> Option B

4. Quick Check:

   Boolean array = True/False values [OK]

Hint: Boolean arrays always hold True or False values [OK]

Common Mistakes:

• Thinking boolean arrays hold numbers or strings
• Confusing boolean arrays with integer arrays

2. Which of the following is the correct way to create a boolean array from a numpy array arr to check where elements are greater than 5?

easy

A. bool_arr = arr > 5

B. bool_arr = arr == 5

C. bool_arr = arr >= 5

D. bool_arr = arr < 5

Solution

1. Step 1: Understand the condition for boolean array

   We want True where elements are greater than 5, so the condition is arr > 5.

2. Step 2: Match the correct syntax

   bool_arr = arr > 5 uses arr > 5, which is correct. arr >= 5 checks greater than or equal to 5, arr == 5 checks equality to 5, and arr < 5 checks less than 5.

3. Final Answer:

   bool_arr = arr > 5 -> Option A

4. Quick Check:

   Condition for > 5 is arr > 5 [OK]

Hint: Use comparison operators directly on arrays for boolean arrays [OK]

Common Mistakes:

• Using == instead of > for greater than
• Confusing >= with >
• Using wrong comparison operator

3. What is the output of the following code?

import numpy as np
arr = np.array([2, 7, 4, 9])
bool_arr = arr < 5
print(bool_arr)

medium

A. [True False True False]

B. [False True False True]

C. [True True True True]

D. [False False False False]

Solution

1. Step 1: Evaluate the condition arr < 5 for each element

   Elements: 2 < 5 is True, 7 < 5 is False, 4 < 5 is True, 9 < 5 is False.

2. Step 2: Form the boolean array

   The boolean array is [True, False, True, False].

3. Final Answer:

   [True False True False] -> Option A

4. Quick Check:

   Check each element < 5 = [True False True False] [OK]

Hint: Check each element against condition to form True/False array [OK]

Common Mistakes:

• Mixing up True and False values
• Forgetting to use < operator correctly

4. Identify the error in this code that tries to create a boolean array for elements equal to 10:

import numpy as np
arr = np.array([10, 5, 10, 3])
bool_arr = arr = 10
print(bool_arr)

medium

A. Array should be a list, not np.array

B. Missing import statement for numpy

C. Using '=' instead of '==' for comparison

D. print statement syntax error

Solution

1. Step 1: Check the comparison operator

   The code uses arr = 10 which assigns 10 to arr, not compares.

2. Step 2: Correct operator for comparison

   To compare elements to 10, use arr == 10 to create a boolean array.

3. Final Answer:

   Using '=' instead of '==' for comparison -> Option C

4. Quick Check:

   Comparison needs '==' not '=' [OK]

Hint: Use '==' for comparison, '=' is assignment [OK]

Common Mistakes:

• Using single '=' instead of '=='
• Confusing assignment with comparison

5. Given a numpy array data = np.array([3, 8, 1, 6, 0]), how can you create a boolean array that is True for elements greater than 2 and less than 7?

hard

A. bool_arr = (data > 2) && (data < 7)

B. bool_arr = data > 2 or data < 7

C. bool_arr = data > 2 and data < 7

D. bool_arr = (data > 2) & (data < 7)

Solution

1. Step 1: Understand element-wise logical operations in numpy

   Use bitwise operators & and | for element-wise AND/OR, not 'and' or 'or'.

2. Step 2: Apply correct syntax for combined condition

   bool_arr = (data > 2) & (data < 7) uses (data > 2) & (data < 7), which is correct for element-wise AND.

3. Final Answer:

   bool_arr = (data > 2) & (data < 7) -> Option D

4. Quick Check:

   Use & for element-wise AND in numpy [OK]

Hint: Use & for element-wise AND, not 'and' [OK]

Common Mistakes:

• Using 'and' instead of '&' for element-wise AND
• Using '&&' which is invalid in Python
• Using 'or' instead of '&' for AND condition