使用dplyr::mutate高效添加函数返回列表的所有元素

嘿，这个问题我熟！刚好可以用dplyr的tidyeval语法来实现，既满足只调用一次testFun，又能轻松给数据框添加新列。

首先，你的testFun其实已经天然支持向量输入（R本身就是向量化语言嘛）——当你把整个x和z向量传进去时，它会返回一个包含两个向量的命名列表（x2和x3）。接下来只要用mutate()结合*!!!*操作符，把这个列表里的元素直接拆成数据框的新列就行。

具体代码如下：

library(dplyr)

testFun <- function(x,z) { list(x2=x*x + z, x3=x*x*x + z) }
have <- data.frame(x=seq(1:10),y=1,z=0)

# 核心实现：一次调用testFun，展开列表添加新列
want <- have %>% 
  mutate(!!!testFun(x, z))

这样就搞定了！!!!在这里的作用是「解引用」列表，把列表里的每个元素作为单独的列参数传给mutate()，相当于手动写x2 = testFun(x,z)$x2, x3 = testFun(x,z)$x3，但好处是只需要调用一次testFun，完美匹配你“计算逻辑复杂，仅调用一次函数”的需求。

如果后续你的testFun逻辑变得更复杂，只要保证它接收向量输入并返回命名列表（每个列表元素对应要添加的列），这个方法就一直适用。

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