如何在自定义函数中使用tibble中的多个变量？

问题原因 #

你当前代码报错的核心是：在mutate中调用user_defined_function()时没有传入任何参数，而你的函数需要接收一个数据框/tibble对象才能提取列进行计算。

解决方案 #

方案1：直接传递管道中的tibble给函数 #

在dplyr管道中，.代表当前传递的数据对象，把它传给你的函数即可：

library(tidyr)
library(dplyr)

rgb_val <- seq(from=0, to=255, by=10)
rgb_grid <- expand_grid(r1 = rgb_val, g1 = rgb_val, b1 = rgb_val,
                        r2 = rgb_val, g2 = rgb_val, b3 = rgb_val)

user_defined_function <- function(x) {
  x[,1] + x[,2] + x[,3] + x[,4] + x[,5] + x[,6]
}

# 修改调用方式，传递当前tibble
rgb_grid %>% mutate(new_cols = user_defined_function(.))

方案2：重写函数适配dplyr列名调用风格（推荐） #

这种方式更符合dplyr的语法习惯，可读性更强，尤其适合复杂逻辑的函数：

# 重写函数，直接接收列作为参数
user_defined_function <- function(r1, g1, b1, r2, g2, b3) {
  r1 + g1 + b1 + r2 + g2 + b3
}

# 在mutate中直接传入对应列名
rgb_grid %>% mutate(new_cols = user_defined_function(r1, g1, b1, r2, g2, b3))

方案3：处理整行数据的通用写法 #

如果你的实际函数需要处理整行的多列数据，可以结合rowwise()和c_across()实现：

# 保留原函数逻辑（接收向量输入）
user_defined_function <- function(x) {
  sum(x) # 这里用sum简化，实际替换为你的复杂逻辑
}

# 按行处理，将指定列打包成向量传给函数
rgb_grid %>%
  rowwise() %>%
  mutate(new_cols = user_defined_function(c_across(r1:b3))) %>%
  ungroup()

补充说明 #

• 方案2是最推荐的写法，因为它明确关联了函数参数和tibble的列，逻辑清晰，便于维护。
• 如果你的实际函数逻辑涉及行内多列的复杂运算，方案3的rowwise()+c_across()组合可以灵活指定需要传入的列（比如c_across(starts_with("r"))匹配所有r开头的列）。

内容的提问来源于stack exchange，提问作者Sushil Kumar Prem