基于第二列条件替换NA值：编写if_else语句求助

需求说明 #

当数据中B列单元格的值等于5时，将对应行的A列单元格设为NA，具体示例如下：

示例数据

期望输出

R 实现（使用tidyverse的if_else函数） #

使用dplyr::mutate结合if_else函数完成条件替换，注意数值型列要使用NA_real_匹配类型：

library(tidyverse)

# 构造示例数据
df <- tibble(
  A = c(1, 4, 7, 10),
  B = c(3, 2, 5, 5)
)

# 执行条件替换
df_processed <- df %>%
  mutate(A = if_else(B == 5, NA_real_, A))

# 查看结果
print(df_processed)

Python 实现（使用pandas） #

可以用numpy.where实现类似if_else的逻辑，或者直接用pandas索引定位修改：

import pandas as pd
import numpy as np

# 构造示例数据
df = pd.DataFrame({
    'A': [1, 4, 7, 10],
    'B': [3, 2, 5, 5]
})

# 方法1：用np.where实现条件替换
df['A'] = np.where(df['B'] == 5, np.nan, df['A'])

# 方法2：用loc索引直接修改（更直观）
# df.loc[df['B'] == 5, 'A'] = np.nan

# 查看结果
print(df)

内容的提问来源于stack exchange，提问作者pradoer