如何对DataFrame每行应用多参数函数并输出DataFrame列表

示例准备 #

你提供的初始示例数据如下：

a <- data.frame(a= c(1,2), b=c(2,3), c=c(0,0))

目标输出为每个行计算结果单独存为DataFrame，列表元素名为r+行号的结构：

b <- list(r1=data.frame(3), r2=data.frame(5))

基础R实现 #

优先选择按行索引遍历的方式，避免apply函数强制转换矩阵带来的类型错误问题：

1. 先定义每行要执行的自定义函数，参数为单行数据：

# 示例函数：对传入的单行数据的a、b、c列求和，返回DataFrame格式结果
row_calc <- function(single_row) {
  sum_val <- sum(single_row$a, single_row$b, single_row$c)
  return(data.frame(sum_val))
}

2. 遍历所有行执行函数，并命名列表：

# 遍历行号执行函数
result_list <- lapply(1:nrow(a), function(i) row_calc(a[i, ]))
# 给列表元素设置对应行号的名称
names(result_list) <- paste0("r", 1:nrow(a))

tidyverse（purrr包）实现 #

用purrr的pmap系列函数可以更简洁的实现按行多参数映射，避免类型转换问题：

1. 加载依赖包：

library(purrr)

2. 直接按行映射执行函数，自动把每行的列对应为函数参数：

result_list <- pmap(a, function(a, b, c) {
  sum_val <- sum(a, b, c)
  return(data.frame(sum_val))
}) %>% 
  # 批量设置列表元素名称
  set_names(paste0("r", seq_along(.)))

结果验证 #

执行后result_list结构和你要求的目标输出完全一致，你可以用str(result_list)查看结构确认。

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