Question

How to use numpy.where() to get elements from two arrays respectively depend on condition in Python

Answer 1

import numpy as np

arr = np.array([45, 13, 18])

result = np.where(arr < 20, [1, 2, 3], [4, 5, 6])
 
print(result)
  
 
  
'''
run:
 
[4 2 3]

'''

 

70+ SQL courses for beginners and professionals

Answer 2

import numpy as np

arr = np.array([3, 30, 18])

result = np.where(arr > 19, [1, 2, 3], [4, 5, 6])
 
print(result)
  
 
  
'''
run:
 
[4 2 6]

'''

 

70+ SQL courses for beginners and professionals