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 vote281 views1 answer
Python data structures 9.4
Python
1 vote245 views1 answer
How to filter queryset with foreign key in django
Python
1 vote248 views1 answer
How to set background color of an image to transparent in pygame
Python
0 vote293 views1 answer
WSGIPassAuthorization on
Python
-1 vote277 views2 answers
Np.apply_along_axis third dimension python
Python
0 vote222 views2 answers
Pandas profile report
Python
0 vote254 views1 answer
How to make python print 2 line text in one code
Python
4 votes244 views3 answers
Txt to list python
Python
0 vote254 views1 answer
Pandan jaya lrt
Python
46 votes257 views4 answers
Rename file python
Python