如何根据条件和另一行的值将 function 应用于 dataframe 行？

Question

If I have a pandas dataframe such as:

a   b   c  
1   2   3 
1   2  -3
2   3   2
4   2  -1

How do change the values of column b based on if the values in c are positive or negative, and use the values in b and a in the operation.

I want to run something like this on each row:

   if (c >= 0):
     b = a - b
   else:
     b = b - a 

and get the dataframe:

a   b   c  
1  -1   3 
1   1  -3
2  -1   2
4  -2  -1

Answer 1

You could use numpy.where which is similar to if/else and is usually faster:

  df.assign(b=np.where(df.c.ge(0), df.a - df.b, df.b - df.a))

    a    b   c
0   1   -1   3
1   1    1  -3
2   2   -1   2
3   4   -2  -1

Alternatively, you could use pandas' where method, which offers a similar approach:

 df.assign(b=df.a.sub(df.b).where(df.c.ge(0), df.b - df.a))

    a    b   c
0   1   -1   3
1   1    1  -3
2   2   -1   2
3   4   -2  -1

Answer 2

You can get the same result with pandas.DataFrame.apply :

df['b'] = df.apply(lambda x: x.a - x.b if x.c >= 0 else x.b - x.a, axis = 1)
#   a  b  c
#0  1 -1  3
#1  1  1 -3
#2  2 -1  2
#3  4 -2 -1