How to use boolean and math expression on numpy array elements in Python

2 Answers

0 votes
import numpy as np
 
arr = np.array([2, 4, 7, 8, 12, 19])
print(arr[arr%2==0]**2)
               

 
'''
run:
 
[  4  16  64 144] 
 
'''

 



answered Jan 25, 2019 by avibootz
0 votes
import numpy as np
 
arr = np.array([2, 4, 7, 8, 12, 19, 27])
print(arr[arr%3==0])
               

 
'''
run:
 
[12 27]
 
'''

 



answered Jan 25, 2019 by avibootz
...