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

0 vote119 views1 answer
Python program using for for the fibonacci number
Python
1 vote99 views1 answer
Create np nan array
Python
1 vote110 views1 answer
Accessing index of dataframe python
Python
34 votes106 views6 answers
How to update a python package
Python
1 vote110 views1 answer
Pythonic
Python
0 vote114 views1 answer
Django add to database
Python
4 votes107 views1 answer
Python make comparison non case sensitive
Python
-1 vote107 views4 answers
How to say hello world in c++
Python
7 votes126 views2 answers
Pygame rotate image
Python
3 votes109 views2 answers
How to make python code run faster
Python