基于多个条件的新列

Question

     a    b
0  100   90
1   30  117
2   90   99
3  200   94

我想用下一个条件创建一个新的df["c"]：

• 如果a > 50 和b 进入(a ± 0.5a)，那么c = a
• 如果a > 50 和b 不在(a ± 0.5a)，那么c = b
• 如果a <= 50，那么*c = a*

输出应该是：

     a    b    c
0  100   90  100
1   30  117   30
2   90   99   90
3  200   94   94

我试过了：

df['c'] = np.where(df.eval("0.5 * a <= b <= 1.5 * a"), df.a, df.b)

但我不知道如何在这句话中包含最后一个条件(If a <= 50, then c = a)。

Answer 1

你就快到了，你只需要在你的 eval 字符串中添加一个or 子句。

np.where(df.eval("(0.5 * a <= b <= 1.5 * a) or (a <= 50)"), df.a, df.b)
#                                           ~~~~~~~~~~~~
array([100,  30,  90,  94])