新列中的熊猫最大或最小行值基于另一个列的值

时间:2019-03-08 18:01:09

标签: python pandas

所以熊猫数据框中有3列

我要创建第4列D,它基于以下条件:如果A> 5,则D的值为B,否则值为C。这是针对每一行

A,B,C    
1,1,2
6,2,3
7,3,1

例如,如果您看到新列D

D row1 = 1#,因为将选择A <5,B的行

D row2 = 3#因为A> 5,所以将选择C行

D row3 = 1#,因为将选择A> 5,所以将选择C行

所以最终的数据帧就像

A,B,C,D    
1,1,2,1
6,2,3,3
7,3,1,1

1 个答案:

答案 0 :(得分:1)

我认为您可以使用

import numpy as np
df['D'] = np.where(df['A']>5, df['C'], df['B'])