循环遍历DataFrame行时，如何将列表结果追加为新列？（Base R与Tidyverse对比）

需求说明：需要将逐行调用dummy_func得到的列表结果，作为多列追加到DataFrame中，且不能重写实际业务函数，只能直接调用。以下是两种实现方案：

Base R 实现 #

利用原生apply函数替代显式for循环，逐行处理后合并结果：

# 逐行调用dummy_func，返回的向量转置后作为新列矩阵
new_cols_matrix <- t(apply(df, 1, function(row) {
  dummy_func(row["X"], row["Y"], vals)
}))

# 为新列命名（V1/V2/V3...）
colnames(new_cols_matrix) <- paste0("V", seq_along(vals))

# 合并原DataFrame与新列
df_base <- cbind(df, new_cols_matrix)

print(df_base)

逻辑说明 #

1. apply(df, 1, ...)：按行遍历DataFrame，把每行数据传入匿名函数；
2. 匿名函数提取当前行的X和Y值，调用dummy_func得到长度等于vals的结果向量；
3. 用t()转置结果矩阵（默认apply返回的是行向量，转置后变成列）；
4. 为新列命名后，通过cbind合并到原DataFrame。

Tidyverse 实现 #

使用dplyr的pmap（多参数向映射）+tidyr的unnest_wider实现，代码更简洁灵活：

library(tidyverse)

df_tidy <- df %>%
  # 逐行传入X/Y，调用dummy_func得到列表列
  mutate(new_vals = pmap(list(X, Y), ~dummy_func(..1, ..2, vals))) %>%
  # 把列表列展开为多列
  unnest_wider(new_vals) %>%
  # 重命名列为V1/V2/V3格式
  rename_with(~paste0("V", seq_along(vals)))

print(df_tidy)

逻辑说明 #

1. pmap(list(X, Y), ...)：将X和Y作为参数列表，逐行传入匿名函数；..1/..2分别指代第一个和第二个参数（即X和Y）；
2. mutate生成的new_vals是一个列表列，每个元素是dummy_func返回的向量；
3. unnest_wider把列表列拆分为多列，自动生成临时列名后，用rename_with统一改为V1/V2/V3格式；
4. 全程使用管道流，逻辑连贯，可读性强。

如果需要更直观的逐行写法（适合固定长度的vals），也可以用rowwise：

df_tidy_rowwise <- df %>%
  rowwise() %>%
  mutate(
    V1 = dummy_func(X, Y, vals[1]),
    V2 = dummy_func(X, Y, vals[2]),
    V3 = dummy_func(X, Y, vals[3])
  ) %>%
  ungroup()

方案对比 #

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