R语言数据框如何在指定列存在时执行两列相减的mutate操作

核心逻辑 #

你需要的mutate if exists效果可以通过计算前先校验依赖列是否存在于当前数据框实现：依赖列齐全才执行减法生成新列，缺任意依赖列就跳过该计算项，不抛出报错。

固定计算项直接实现 #

如果只需要计算固定的几组差值，直接在mutate()内加列存在判断即可，可直接替换原有代码：

library(dplyr)

Df <- df %>% 
  mutate(
    # 校验x差值的两个依赖列都存在才计算，否则跳过
    columnxdif = if (all(c("columnxbeg", "columnxend") %in% names(.))) {
      columnxbeg - columnxend
    } else {
      NULL
    },
    # 校验y差值的两个依赖列都存在才计算，否则跳过
    columnydif = if (all(c("columnybeg", "columnyend") %in% names(.))) {
      columnybeg - columnyend
    } else {
      NULL
    }
  )

代码里的.代表当前传入mutate的数据流，all(...)判断会严格检查依赖列是否都在数据框列名中；如果依赖列不全，赋值NULL会让mutate自动跳过该新列生成，不会触发找不到对象的报错。

多计算项批量实现 #

如果需要生成的差值列较多，可以封装通用判断逻辑，避免重复写判断代码：

# 封装安全计算差值的通用函数
safe_calc_diff <- function(beg_col, end_col, new_col) {
  ~if (all(c(beg_col, end_col) %in% names(.))) {
    setNames(list(.data[[beg_col]] - .data[[end_col]]), new_col)
  } else {
    NULL
  }
}

# 批量传入需要计算的列组合即可
Df <- df %>%
  mutate(
    !!!list(
      safe_calc_diff("columnxbeg", "columnxend", "columnxdif"),
      safe_calc_diff("columnybeg", "columnyend", "columnydif")
    )
  )

效果说明 #

如果数据框中缺失columnxend列，上述代码不会中断运行，会正常完成columnydif的计算并把该列添加到数据框中，完全匹配预期效果。

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