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 vote354 views1 answer
Couchbase python
Python
12 votes388 views6 answers
How to print a variable in python
Python
0 vote338 views1 answer
Print a line using python, All the word lengths should be fixed i.e., every word should have the width of the longest word
Python
41 votes301 views6 answers
How to flatten list of lists in python
Python
0 vote371 views1 answer
How to customize simplejwt error response message in django restframework
Python
2 votes335 views3 answers
Pandas dataframe compare two dataframes and extract difference
Python
2 votes395 views3 answers
Streamlit pip
Python
0 vote333 views1 answer
Colorbar remove tick lines and border
Python
1 vote351 views1 answer
Par o inpar python
Python
0 vote359 views1 answer
Pandas df to R df
Python