所以熊猫数据框中有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
答案 0 :(得分:1)
我认为您可以使用
import numpy as np
df['D'] = np.where(df['A']>5, df['C'], df['B'])