python - 简化pan​​das表达式

Question

这不是一个好问题，但问题是这样的:
我正在做一个非常简单的计算，结合了 DataFrame 的几列，但前提是其中一列具有特定值。想法是:

if df.x==1:
    df.y = df.y - df.a/df.b

现在循环当然很慢，所以我在 DataFrame 的子集上执行此操作，但这很快就会变得非常冗长:

df.loc[df.x==1, 'y'] = df.loc[df.x==1, 'y'] - df.loc[df.x==1, 'a']/df.loc[df.x==1, 'b']

我感觉有更好的方法可以做到这一点，有什么想法吗？

Answer 1

正如 @EdChum 在他的评论中提到的，您可以直接使用 .loc 。您可以使用 -= 表示法进一步简化。

df = pd.DataFrame({'x': [1, 2, 3], 
                   'y': [1, 2, 3], 
                   'a': [1, 2, 3], 
                   'b': [2, 2, 2]})

>>>  df
   a  b  x  y
0  1  2  1  1
1  2  2  2  2
2  3  2  3  3

df.loc[df.x==1, 'y'] -= df.a / df.b

>>> df
   a  b  x    y
0  1  2  1  0.5
1  2  2  2  2.0
2  3  2  3  3.0