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 vote182 views1 answer
Prints out the schema in the tree format
Python
0 vote161 views1 answer
Python dictionary get keys with condition on value
Python
0 vote173 views1 answer
Use a csv file on internet as an api in python
Python
0 vote174 views1 answer
Python load file with multiple jsons
Python
0 vote179 views1 answer
Python remove punctuation from a string
Python
0 vote163 views2 answers
Python create adictionary randomly assigning clors to categorical vairables
Python
0 vote172 views1 answer
Tensorflow turn off gpu
Python
0 vote172 views2 answers
Wxpython menu callback stackoverflow
Python
0 vote185 views1 answer
Django array of dates
Python
2 votes163 views1 answer
Use of kwargs and args in python classes
Python