如何用dplyr管道将R函数返回的列表存入数据框的列表列？

我有几个返回列表形式结果的R函数，想要把这些列表输出存入包含函数输入变量的数据框中。示例代码如下：

library(dplyr)
# 定义函数
testFunc <- function(a){
  a = a
  b = a+1
  c = list(out1=a, out2=b)
  return(c)
}
# 测试数据
dat <- data.frame(x=1:5)

# 尝试直接存入，会报错
datProcessed <- dat %>%
  mutate(calcd = testFunc(x))

运行上述代码会报错：calcd must be size 10 or 1, not 2。但如果只提取列表中的单个元素，就能正常运行：

datProcessed <- dat %>%
  mutate(calcd = testFunc(x)$out2)

下面提供两种可行的解决方法：

方法1：使用rowwise()配合list() #

默认情况下mutate会按向量批量处理数据，导致函数返回的列表被拆解成多个元素，引发维度不匹配问题。用rowwise()让数据按行处理，再用list()把函数返回的列表包裹成单个元素存入列中：

datProcessed <- dat %>%
  rowwise() %>%
  mutate(calcd = list(testFunc(x))) %>%
  ungroup() # 处理完记得取消行分组，避免影响后续操作

方法2：使用purrr::map() #

purrr包的map()函数会逐个对x的元素调用testFunc，返回的结果本身就是一个列表，刚好可以直接作为数据框的列表列：

library(dplyr)
library(purrr)

datProcessed <- dat %>%
  mutate(calcd = map(x, testFunc))

后续操作：展开列表列 #

如果需要把列表列中的元素展开成单独的列，可以用unnest_wider()：

datProcessed %>%
  unnest_wider(calcd)

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