Python 基于其他列从列中获取值

我想从同一df的另一列中提取具有条件的数据帧的值

    A    B
0   2    10
1   3    60
2   7    50
3   1    100

我只想从

B

中获取

A

values>2，并用固定值=5填充从

B

中获取的值

像这样：

    A    B
0   2    5
1   3    60
2   7    50
3   1    5

使用：

---

使用熊猫系列，其中：

df["B"] = df["B"].where(df["A"].gt(2), 5)
print(df)

输出：

   A   B
0  2   5
1  3  60
2  7  50
3  1   5

   A   B
0  2   5
1  3  60
2  7  50
3  1   5