從基於其他列的列中獲取值

Question

我想從同一 df 的另一列中提取具有條件的數據框的值。

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

我只想從B獲取A值 > 2，並用固定值 = 5 填充我從B中獲取的值

像這樣：

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

Answer 1

用：

import numpy as np
df['B'] = np.where(df['A']>2,df['B'],5)
df
   A   B
0  2   5
1  3  60
2  7  50
3  1   5

Answer 2

使用pandas.Series.where ：

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