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 vote131 views1 answer
Loop per n (batch)
Python
0 vote161 views1 answer
Python size of linked list
Python
1 vote133 views1 answer
Jupyter notebook "TypeError: 'numpy.ndarray' object is not callable"
Python
1 vote186 views2 answers
ValueError: Cannot specify ',' with 's'.
Python
1 vote130 views1 answer
Count missing values groupby
Python
0 vote152 views1 answer
Python click activator
Python
8 votes140 views1 answer
Python pdf to image
Python
0 vote141 views1 answer
Figure in matplotlib
Python
0 vote153 views1 answer
Python mysqlclient not installing
Python
23 votes282 views6 answers
Flask post
Python Flask