如何简化R语言dataframe内数学运算的列变量引用写法

你需要的免data$前缀、直接写列名公式的需求完全可以实现，不需要借助apply类函数，有更简洁的向量化实现方案：

• base R原生方案（无额外依赖） #

  使用with()函数即可直接在数据框的上下文环境中执行运算，不需要单独提取变量，完全匹配你给出的公式写法：

  # 运算结果返回为独立向量
  result <- with(data, sqrt(var2)*(var4+var1)/var3 + var5/log(var3))
  

  如果需要将计算结果作为新列存回原数据框，可使用transform()函数：

  # 计算结果作为new_col列存入原数据框
  data <- transform(data, new_col = sqrt(var2)*(var4+var1)/var3 + var5/log(var3))
  

• tidyverse生态方案 #

  如果你日常使用dplyr进行数据处理，mutate()函数同样支持直接写列名执行运算：

  library(dplyr)
  data <- data %>%
    mutate(new_col = sqrt(var2)*(var4+var1)/var3 + var5/log(var3))
  

补充说明 #

apply类函数多用于批量对多列/多行执行相同的标准化运算，你的需求是自定义公式的逐元素向量化计算，上述方案代码更简洁，执行效率也更高，无需额外使用apply族函数实现。

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