R语言DataFrame如何根据某列字符串匹配提取对应列值生成新列

针对你需要根据co列存储的列名，提取对应列的行值生成新列的需求，以下是几种可落地的实现方案，优先推荐无依赖、性能最优的基础R写法：

• 基础R矩阵索引法（首选）
  基于R原生的数据框矩阵索引规则实现，不需要加载任何第三方包，无论co列有多少种取值、数据集规模多大都能稳定运行，代码如下：

  # 构造示例数据
  rrr <- data.frame(a=c(1,2,3), b=c(3,4,5), co=c('a','b','a'))
  # 动态匹配生成新列
  rrr$newColumn <- rrr[cbind(1:nrow(rrr), rrr$co)]
  

  运行后得到的结果完全匹配预期：

  a b co newColumn
  1 1 3  a         1
  2 2 4  b         4
  3 3 5  a         3
  

  注意：不建议使用apply逐行遍历的方式实现该需求，apply会先将数据框所有列统一转换为字符型再运算，容易引发数据类型错误，性能也远低于矩阵索引法。

• 条件判断赋值法（仅适配少量固定值场景）
  如果co列的可选取值极少（比如仅存在a、b两个固定值），也可以直接用ifelse做条件判断赋值，但如果后续co列新增其他取值，需要不断嵌套判断逻辑，扩展性很差：

  rrr$newColumn <- ifelse(rrr$co == 'a', rrr$a, rrr$b)
  

• dplyr行级运算法
  如果你日常使用tidyverse生态做数据处理，也可以通过行分组+动态取值的方式实现，注意运算后要取消行分组，避免影响后续数据操作：

  library(dplyr)
  rrr <- rrr %>%
    rowwise() %>%
    mutate(newColumn = get(co)) %>%
    ungroup()
  

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