如何使用df中的多个列运行多个条件来计算新列？[副本]

Question

例如，假设我有一个df

a    b     c       

1    3     5       
5    9     4      

我有if条件：

if a < 2:
   3
elif  a < 3:
   4
else: b + c

我如何测试条件并为我的df的每一行返回一个结果，如下面所示？

a    b     c    d

1    3     5    3 
5    9     4    13

编辑：理想情况下，我想创建一个函数，允许我

def函数（a，b，c）df['d']=函数（a，b，c）

并计算所有行的数据。因为在实际数据中，有100个条件语句和10个列。

Answer 1

您可以使用应用方法：

def custom_calc(x, a , b):
    if x[0] < 2:
        return 3

    if x[0] > 3:
        return x + a + b

df.a = df[['a', 'b', 'c']].apply(func=check, args=(df.b,df.c))

如您所愿编辑自定义函数

Answer 2

可以沿着轴1使用应用函数。

def f(row):
    if row['a'] > 2:
        return 3
    elif row['a'] > 3:
        return 4
    else:
        return  row['b']+row['c']


df.apply(f,axis=1)

#output
0    8
1    3
dtype: int64