如何用R扩展数据集以包含前两列的所有组合？

melt 和 pivot_longer 是用于数据长宽格式转换的工具，并不适合生成列的笛卡尔积组合，你可以试试以下几种更直接的方法：

方法1：tidyverse 实现（适合常规场景） #

利用 expand_grid 生成前两列的所有可能组合，再通过左连接补全原数据的其他列：

library(tidyverse)

# 提取前两列的唯一值并生成所有组合
all_combinations <- df %>%
  select(1:2) %>%
  distinct() %>%
  expand_grid(col1 = unique(.$col1), col2 = unique(.$col2))

# 左连接原数据，补全所有组合对应的其他字段（缺失值用NA填充）
expanded_df <- left_join(all_combinations, df, by = c("col1", "col2"))

方法2：Base R 原生实现（无需额外包） #

用原生函数 expand.grid 生成笛卡尔积，再通过 merge 合并数据：

# 提取前两列的唯一值
unique_col1 <- unique(df[[1]])
unique_col2 <- unique(df[[2]])

# 生成所有组合
all_combinations <- expand.grid(col1 = unique_col1, col2 = unique_col2, stringsAsFactors = FALSE)

# 合并原数据，保留所有组合
expanded_df <- merge(all_combinations, df, by = c("col1", "col2"), all.x = TRUE)

方法3：data.table 实现（适合大数据集） #

如果你的数据集量级很大（数千条以上），data.table 的效率会更高：

library(data.table)

setDT(df)

# 生成前两列的所有组合
all_combinations <- CJ(col1 = unique(df[[1]]), col2 = unique(df[[2]]))

# 合并数据，保留所有组合
expanded_df <- merge(all_combinations, df, by = c("col1", "col2"), all.x = TRUE)

以上方法都会生成前两列的所有可能组合，原数据中不存在的组合对应的其他列会用 NA 填充，完全满足扩展数据集的需求。

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