Question

How to use NumPy where() with multiple conditions in Python

Answer 1

import numpy as np
 
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
 
condition1 = arr % 2 == 0  # even numbers
condition2 = arr > 4      # numbers greater than 4
 
# Apply conditions
arr = np.where(condition1 & condition2, arr, 0)

# If the element satisfies both conditions → keep the original value
# Otherwise → replace it with 0
 
print(arr)
 
 
  
'''
run:
  
[ 0  0  0  0  0  6  0  8  0 10]
  
'''

 

70+ SQL courses for beginners and professionals