Pandas where

In [20]: df
Out[20]:
   A  B  C
0  1  2  1
1  2  3  0
2  3  4  0
3  4  5  1

# Note that df.C is a mask
# Note that this is np.where, not df.where. That is different.
In [21]: df['D'] = np.where(df.C, df.A, df.B)

In [22]: df
Out[22]:
   A  B  C  D
0  1  2  1  1
1  2  3  0  3
2  3  4  0  4
3  4  5  1  4

Source: stackoverflow.com

Similar codes

2 votes150 views1 answer
Pyspark convert float results to integer replace
Python
8 votes78 views4 answers
Python gui
Python
26 votes123 views10 answers
For loop
Python
2 votes141 views1 answer
Python how to unnest a nested list
Python
3 votes155 views2 answers
Django postgres connection
Python Django
2 votes124 views2 answers
Pyinstaller exe version info
Python
0 vote136 views1 answer
Pytorch squeeze
Python
4 votes157 views1 answer
Edit line if str end with pandas
Python
21 votes142 views2 answers
How to update python
Python
9 votes114 views1 answer
Cv2.cvtcolor grayscale
Python