Numpy where

import numpy as np

# Return elements chosen from x or y depending on condition.
a = np.arange(10) # [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
single_where = np.where((a<5),-1,a) # [-1, -1, -1, -1, -1,  5,  6,  7,  8,  9]
multiple_where = np.where((a<5),-1,np.where((a>5),0,a)) # [-1, -1, -1, -1, -1,  5,  0,  0,  0,  0]

Similar codes

0 vote200 views1 answer
Connect mysql sql alchemy
Python
0 vote220 views1 answer
Free the Bunny Prisoners
Python
0 vote193 views1 answer
Tf.slice
Python
2 votes187 views1 answer
Attributes in python
Python
1 vote191 views1 answer
Find nth root of m using python
Python
0 vote209 views1 answer
Extract month as integer python
Python
1 vote196 views1 answer
How to remove python3 on mac
Python
30 votes195 views5 answers
Colab mount drive
Python
4 votes211 views2 answers
How to determine python project parent dir
Python
1 vote222 views2 answers
Run flask in background
Python Flask