在Pandas DataFrame中，如何仅在列为1时填充

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [0, 1, 0, 1, 0],
        'B': [1, 2, 3, 4, 5]}
df = pd.DataFrame(data)

# 仅在列A为1时填充
df['B'] = df['B'].fillna(value=10, mask=df['A'] == 1)

print(df)

   A     B
0  0   1.0
1  1   2.0
2  0   3.0
3  1  10.0
4  0   5.0