如何用dplyr的mutate批量处理相似命名列计算delta_i？

现有如下数据：

data <- tribble(
  ~a1, ~a2, ~b1, ~b2, ~c1, ~c2,
  32, 32, 50, 12, 12, 50,
  48, 20, 55, 43, 10, 42
)

需要针对i={1,2}计算：
delta_i = (a_i - c_i) / ((a_i + b_i) * c_i + a_i)

直接编写mutate语句会产生大量重复代码，封装delta函数后仍存在重复；尝试循环mutate时出现类型错误，且循环方式不够优雅。期望找到利用across的简洁写法，用一行mutate完成批量计算，实现类似mutate("delta{i}" := delta(a{i}, b{i}, c{i}))的效果。

可以利用dplyr的across结合字符串提取和列匹配实现需求，核心是通过列名后缀关联对应组的列，批量计算并生成目标列：

library(dplyr)
library(stringr)

data <- data %>%
  mutate(across(starts_with("a"), 
                ~ (. - get(str_replace(cur_column(), "a", "c"))) / ((. + get(str_replace(cur_column(), "a", "b"))) * get(str_replace(cur_column(), "a", "c")) + .),
                .names = "delta{str_remove(.col, 'a')}"))

代码说明 #

• starts_with("a")：选中所有以a开头的列（a1、a2）作为遍历对象
• str_replace(cur_column(), "a", "c")：将当前列名中的a替换为c，获取对应组的c列（如a1对应c1），同理替换为b获取对应b列
• .names = "delta{str_remove(.col, 'a')}"：将生成的新列命名为delta加上原列名去掉a后的部分（a1对应delta1，a2对应delta2）

运行后数据会新增delta1和delta2列，结果示例：

# A tibble: 2 × 8
     a1    a2    b1    b2    c1    c2 delta1 delta2
  <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>  <dbl>  <dbl>
1    32    32    50    12    12    50  0.052  -0.03
2    48    20    55    43    10    42  0.077  -0.02

如果想让逻辑更清晰，也可以先封装计算函数：

calc_delta <- function(a_col, suffix) {
  b_col <- get(paste0("b", suffix))
  c_col <- get(paste0("c", suffix))
  (a_col - c_col) / ((a_col + b_col) * c_col + a_col)
}

data <- data %>%
  mutate(across(starts_with("a"), 
                ~ calc_delta(., str_remove(cur_column(), "a")),
                .names = "delta{str_remove(.col, 'a')}"))

这种写法既避免了重复代码，又保持了可读性，符合简洁性要求。

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