if函数python_python – 在pandas dataframe中的if else函数

你可以使用

numpy.where：

def my_fun (var1,var2,var3):

df[var3]= np.where((df[var1]-df[var2])>0, df[var1]-df[var2], 0)

return df

df1 = my_fun('age1','age2','diff')

print (df1)

age1 age2 diff

0 23 10 13

1 45 20 25

2 21 50 0

错误更好解释here.

使用apply更慢的解决方案,需要axis = 1,按行进行数据处理：

def my_fun(x, var1, var2, var3):

print (x)

if (x[var1]-x[var2])>0 :

x[var3]=x[var1]-x[var2]

else:

x[var3]=0

return x

print (df.apply(lambda x: my_fun(x, 'age1', 'age2','diff'), axis=1))

age1 age2 diff

0 23 10 13

1 45 20 25

2 21 50 0

也可以使用loc,但有时可以覆盖数据：

def my_fun(x, var1, var2, var3):

print (x)

mask = (x[var1]-x[var2])>0

x.loc[mask, var3] = x[var1]-x[var2]

x.loc[~mask, var3] = 0

return x

print (my_fun(df, 'age1', 'age2','diff'))

age1 age2 diff

0 23 10 13.0

1 45 20 25.0

2 21 50 0.0