R 的 ifelse 语句的 Python 版本

Question

在学习了 R 和一个简单的 ifelse 语句之后，我正在尝试学习 Python。

在 R 中我有:

df$X <- if(df$A == "-1"){-df$X}else{df$X}

但我不确定如何在 Python 中实现它，我已经尝试过:

df['X'][df['A'] <1] = -[df['X']
df['X'][df['A'] >1] = [df['X']

但这会导致错误，希望得到一些帮助。

Answer 1

等价于 np.where :

import numpy as np
np.where(df['A'] < 1, -df['X'], df['X'])

这会检查列中的值是否为 A小于 1。如果是，则返回 df['X'] 中的相应值乘以 -1。 ，否则返回df['X']中对应的值.

也就是说，您的错误/警告可能是由于链式索引而引发的。而不是 df['X'][df['A'] <1]你应该使用 df.loc[df['A'] <1, 'X'] .然后您可以按照问题中所示的两个步骤执行相同的操作。