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 vote331 views1 answer
Add up the elements in this RDD
Python
5 votes327 views2 answers
Pandas convert index to column
Python
0 vote326 views1 answer
Django only certain columns from database
Python
7 votes343 views1 answer
From string to time python dataframe
Python
0 vote325 views1 answer
Django serialize a get object is not iterable
Python
0 vote331 views3 answers
How can I corect word spelling by use of nltk?
Python
0 vote325 views1 answer
How do you make plot show with matplotlib ion method
Python
2 votes294 views2 answers
'flask' is not recognized as an internal or external command, operable program or batch file.
Python
0 vote319 views1 answer
Websocket api python on close
Python
6 votes314 views2 answers
Pygame size of image
Python