如何根據其他列的值將公式應用於 Dataframe 中的列

Question

我有一個 DataFrame，我想根據 B 列在 A 列上應用公式。

如果 B 列有一些值，則在 A 列（ 100 - column value ）上應用公式。 下面是我的數據。

DataFrame 輸入

      A         B 
0     35       TYLER
1     20       MACK
2     30       MACK
3     40       MITCH

如果 B 列有“MACK”和“MITCH”，則應用公式，否則不應用。 我怎樣才能做到這一點？

DataFrame Output

       A         B 
0     35       TYLER
1     80       MACK
2     70       MACK
3     60       MITCH

Answer 1

您可以將公式限制為僅具有 boolean 索引的匹配行：

apply_rows = df["B"].isin(["MACK", "MITCH"])
df.loc[apply_rows, "A"] = 100 - df.loc[apply_rows, "A"]