R语言基于多列匹配规则替换数据框指定列值的方法求解

实现思路 #

基于name1与非空name2的一一对应关系构造映射表，筛选出name4与同行name1相等的行后用映射值批量替换，全程不需要硬编码任何具体取值。

base R 实现方案 #

# 构造name1到非空name2的命名映射向量
name_map <- setNames(df$name2[!is.na(df$name2)], df$name1[!is.na(df$name2)])

# 定位需要替换的行
replace_idx <- df$name4 == df$name1

# 批量替换目标值
df$name4[replace_idx] <- name_map[df$name1[replace_idx]]

运行后得到的结果如下，完全符合预期：

name1 name2 name3 name4
1     a     x  <NA>     x
2     b  <NA>    b1    b1
3     c  <NA>    c1    c1
4     a  <NA>  <NA>     x

tidyverse 实现方案（可选） #

如果你习惯用dplyr语法，可以参考如下写法：

library(dplyr)

df <- df %>%
  left_join(
    df %>% filter(!is.na(name2)) %>% select(name1, name2_map = name2),
    by = "name1"
  ) %>%
  mutate(name4 = if_else(name4 == name1, name2_map, name4)) %>%
  select(-name2_map)

注意事项 #

因为你已经提前对数据做过去重处理，同一个name1不会对应多个不同的非空name2值，上述两种方案都不会出现匹配冲突。如果遇到部分name1没有对应非空name2的场景，替换后会返回NA，你可以根据业务需要额外加判断逻辑处理这类边界情况。

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