R语言如何在多列搜索指定值并提取匹配列左侧对应值

不需要手动逐列嵌套which()或者ifelse()，通过自动匹配待搜索列和其左侧相邻列的对应关系即可实现，列数再多也能通用，提供两种实现方式按需选择：

方法1：易读通用版（适配绝大多数场景） #

代码逻辑简单易修改，适配常规数据量：

# 加载示例数据
a2 <- data.frame(position_1 = c("Senator", "Rep", "Senator", "Rep"),
                 congress_1 = c(107, 107, 108, 109),
                 position_2 = c("Senator", "Rep", "Senator", "Rep"),
                 congress_2 = c(108, 108, 109, 110))

# 配置搜索参数
target_val <- 108
# 自动识别所有待搜索的congress列，无需手动枚举
search_cols <- grep("^congress_", names(a2), value = TRUE)
# 自动匹配每个搜索列左侧对应的取值列（这里是position_x列）
extract_cols <- sub("congress_", "position_", search_cols)

# 计算结果
res <- data.frame(
  row = seq_len(nrow(a2)),
  position_108 = apply(a2, 1, \(row) {
    match_pos <- which(as.numeric(row[search_cols]) == target_val)
    if (length(match_pos) == 0) return("none")
    # 若同一行有多个匹配，默认取第一个匹配对应的左侧列值
    as.character(row[extract_cols[match_pos[1]]])
  })
)

运行后输出结果完全符合预期：

row position_108
1   1      Senator
2   2      Senator
3   3          Rep
4   4         none

方法2：高性能向量化版（适配百万行级大数据量） #

去掉逐行apply循环，用矩阵索引实现，速度比方法1快数倍到数十倍：

target_val <- 108
search_cols <- grep("^congress_", names(a2), value = TRUE)
extract_cols <- sub("congress_", "position_", search_cols)

# 构建匹配布尔矩阵
match_logic <- a2[search_cols] == target_val
res <- data.frame(row = seq_len(nrow(a2)), position_108 = "none")
# 定位有匹配的行，以及每行第一个匹配对应的提取列
have_match <- rowSums(match_logic) > 0
match_extract_col <- extract_cols[max.col(match_logic, ties.method = "first")[have_match]]
# 批量赋值
res$position_108[have_match] <- as.character(
  a2[cbind(which(have_match), match(match_extract_col, names(a2)))]
)

自定义调整说明 #

• 如果列没有统一命名规律，不需要用grep自动识别列名，直接手动给search_cols和extract_cols传对应列名的向量即可，比如search_cols <- c("congress_1","congress_2")，extract_cols <- c("position_1","position_2")
• 如果同一行存在多个匹配项，需要拼接所有匹配到的左侧列值，可以把方法1里的返回值改成paste(as.character(row[extract_cols[match_pos]]), collapse = "、")
• 无匹配时的填充值可以直接修改return("none")或者默认赋值的"none"为你需要的内容

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