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

2 votes195 views1 answer
How to split image dataset into training and test set keras
Python
8 votes192 views1 answer
Tkinter clear entry
Python
1 vote213 views1 answer
Two for loops in list comprehension
Python
0 vote195 views1 answer
Django insert data into database foreign key view.py
Python
0 vote170 views2 answers
Odoo wizard current user login
Python
0 vote214 views1 answer
Dictionary Get All Keys
Python
0 vote234 views1 answer
How to convert linked list to list in python
Python
0 vote199 views1 answer
"serializers" is not defined
Python
0 vote248 views2 answers
Radix sort python
Python
0 vote194 views1 answer
How to find the ascii value of character in python
Python