NumPy where() Multiple Conditions

In NumPy, you can use the where() function to apply multiple conditions and return values based on those conditions. You can specify multiple conditions using logical operators like & (and), | (or), and ~ (not). The numpy.where() is a versatile function that can be used to create new arrays or update existing arrays based on specified conditions. In this article, I will explain how to use the where() function with multiple conditions with examples.

1. Quick Examples of Using where() with Multiple Conditions

If you are in a hurry, below are some quick examples of how to use multiple conditions in where() condition.

# Quick examples of where() with multiple conditions

# Example 1: Get the indices of array elements
arr1=np.where(arr)

# Example 2: Use numpy.where() multiple conditions
# With the & operator
arr2 = arr[np.where((arr >5) & (arr <25))]

# Example 3: Multiple conditions using | operator
# Use numpy.where() function
arr2 = arr[np.where((arr >5) | (arr % 2 == 0))]

# Example 4: Find number of values meet in a condition
arr2 = (arr[np.where((arr >5) | (arr % 2 == 0))]).size

# Example 5: Use numpy.where() multiple conditions 
# Using numpy.logical_and() function
arr2 = arr[np.where(np.logical_and(arr >5, arr <25))]

# Example 6: Use numpy.where() multiple conditions 
# Using logical_or() function
arr2 = arr[np.where(np.logical_or(arr >5, arr % 2 == 0))]

2. Usage of NumPy where()

First, let’s quickly see how to use a where() function, and later I will cover examples of how to use it with multiple conditions. When we pass the NumPy array to the numpy.where() it will return the indices of NumPy array elements. Let’s create a NumPy array using numpy.array().

# Import numpy
import numpy as np

arr = np.array([5, 10, 15, 20, 25])
print("Given array:", arr)

# Get the indices of array elements
arr1=np.where(arr)
print("Get indices of given array:", arr1)

Yields below output.

3. NumPy where() Multiple Conditions With the & Operator

To select the NumPy array elements from the existing array based on multiple conditions use the & operator along with the where() function. You can specify multiple conditions inside the where() function by enclosing each condition inside a pair of parenthesis and using an & operator.

Let’s pass the multiple conditions with the help of the & operator to the where() function and get the selected elements of the given NumPy array. For example,

import numpy as np

# Create a numpy array 
arr = np.array([5, 10, 15, 20, 25])

# Use numpy.where() multiple conditions with the & operator
arr2 = arr[np.where((arr >5) & (arr <25))]
print("Get selected array elements:", arr2)

Yields below output.

4. Use where() Multiple Conditions With the | Operator

You can use the np.where() function with multiple conditions to modify the existing array based on those conditions.

In the below example, you can use the | (bitwise OR) operator to filter elements from a NumPy array. First condition1 checks if the elements arr are greater than 5, and condition2 checks if the elements are divisible by 2. The | operator is used to combine these conditions. This function returns the indices where either condition1 or condition2 is true. Finally, the elements corresponding to the selected indices are extracted from the original array arr and stored in the arr2 array.

import numpy as np

# Create a numpy array 
arr = np.array([5, 10, 15, 20, 25])

# Multiple conditions using | operator
# Use numpy.where() function
arr2 = arr[np.where((arr >5) | (arr % 2 == 0))]
print("Get selected array elements:\n", arr2)

# Output:
# Get selected array elements:
#  [10 15 20 25]

You can also use the size function to simply find how many values meet one of the conditions using (arr[np.where()]).size function. For instance, to create a NumPy array called arr with values [5, 10, 15, 20, 25] and then filters the array based on a condition using boolean indexing. Specifically, it selects elements arr that are greater than 5 or divisible by 2.

The condition arr > 5 checks for elements greater than 5, and arr % 2 == 0 checks for elements divisible by 2. The | operator represents logical OR, so it selects elements that satisfy either of these conditions.

import numpy as np

# Create a numpy array 
arr = np.array([5, 10, 15, 20, 25])

# Find number of values meet in a condition
arr2 = (arr[np.where((arr >5) | (arr % 2 == 0))]).size
print("Get selected array element:", arr2)

# Output:
# Get selected array element: 4

5. Use where() Multiple Conditions With the logical_and()

numpy.logical_and() function is used to calculate the element-wise truth value of AND gate in Python. Using this function inside the where() function to specify multiple conditions and get the selected elements from an existing array.

In this example, np.logical_and(arr > 5, arr < 25) creates a boolean array with True where both conditions are satisfied and False otherwise. np.where() then finds the indices where the conditions are True, and finally, these indices are used to extract the corresponding elements from the original array.

import numpy as np

# Create a numpy array 
arr = np.array([5, 10, 15, 20, 25])

# Use numpy.where() multiple conditions 
# Using numpy.logical_and() function
arr2 = arr[np.where(np.logical_and(arr >5, arr <25))]
print("Get selected array element:", arr2)

# Output:
# Get selected array element: [10 15 20]

6. Use where() Multiple Conditions With the logical_or()

You can use np.where() with np.logical_or() to filter elements based on multiple conditions using logical OR. For instance, np.logical_or(arr > 5, arr % 2 == 0) creates a boolean array with True where either condition (greater than 5 or divisible by 2) is satisfied and False otherwise. np.where() then finds the indices where the conditions are True, and these indices are used to extract the corresponding elements from the original array.

import numpy as np

# Create a numpy array 
arr = np.array([5, 10, 15, 20, 25])

# Use numpy.where() multiple conditions with the .logical_or()
arr2 = arr[np.where(np.logical_or(arr >5, arr % 2 == 0))]
print("Get selected array element:", arr2)

# Output:
# Get selected array element: [ 10 15 20 25]

Frequently Asked Questions NumPy where() Multiple Conditions

Conclusion

In this article, I have explained how to use multiple conditions on NumPy where() function by using the &(and), |(or), logical_and() and, logical_or() functions with examples.

Happy Learning!!

Related Articles

• How to transpose the array?
• Get the sum of two arrays
• NumPy where() Function With Examples
• How to get the power value of array?
• Get the cumulative sum of array
• Python NumPy zeros() Function
• How to Use NumPy Sum() in Python
• How to Append NumPy Arrays Examples
• How to delete columns & rows of NumPy array?
• How to Use NumPy random.normal() In Python?

References

• https://np.org/doc/stable/reference/generated/np.where.html