python - pandas 中的条件方法链接

Question

在 pandas 中使用方法链时，是否有一种简单的通用方法可以使方法以 if 语句为条件？

模拟示例:

df = pd.DataFrame({'A':['one', 'two'], 'B':['one', 'two']})

change_to_numeric = False

df = (df
    .query("A == 'one'")
    .replace('one', 1)         # <-- Execute this row only "if change_to_numeric == True"
)

谢谢!

Answer 1

您可以使用pipe :

df = pd.DataFrame({'A':['one', 'two'], 'B':['one', 'two']})

change_to_numeric = False

df = (df
    .query("A == 'one'")
    .pipe(lambda d: d.replace('one', 1) if change_to_numeric else d)
)

change_to_numeric = False 的输出:

     A    B
0  one  one

change_to_numeric = True 的输出:

   A  B
0  1  1