仅用R基础包与dplyr实现pivot_wider的替代方案

问题背景 #

需要生成桑基图所需的宽表，tidyr::pivot_wider是适配工具，但运行环境无权限安装额外库，仅能使用R基础包与dplyr。现有数据框df1：

df1 <- structure(list(SEZ = c("1_1_1", "1_1_1", "1_1_2", "1_1_3", "1_1_3", 
                              "1_1_4", "1_1_5", "1_2_1", "1_2_1"),
                      Class = c(1L, 5L, 5L, 1L,   5L, 1L, 2L, 1L, 5L), 
                      Val = c("AB", "AB", "AB", "B", "AB", "B", "CD", "CD", "AB")), 
                 row.names = c(NA, -9L), class = "data.frame")

参考方案时因Val列为文本类型而非数值报错，已实现临时解决方法，现寻求更简洁的替代方案。

临时解决方法（原方案） #

# 将文本列Val转为数值，后续再转回文本（R 4.3.1 & dplyr 1.1.2）
# 创建映射列表
list_mapTclass <- seq(1:length(unique(df1$Val)))
names(list_mapTclass) <- unique(df1$Val)

# 生成数值索引列
df1$idxVal <- sapply(df1$Val, function(x) list_mapTclass[[x]] )

# 生成宽表矩阵
wide_df_matrix <- xtabs(idxVal ~ SEZ + Class, df1)

# 转换为数据框并重命名列
wide_df0 <- as.data.frame.matrix(wide_df_matrix) %>% 
  setNames(paste0('col_', names(.)))

# 创建反向映射列表
list_mapTclassInverted <- setNames(names(list_mapTclass), list_mapTclass)
# 可选：为0添加映射（若存在0值）
# list_mapTclassInverted[['0']] <- 'Unknown'

# 将数值转回文本
wide_df00 <- wide_df0
wide_df00[] <- list_mapTclassInverted[unlist(wide_df00)]

# 添加SEZ列并重置行名
wide_df0$SEZ <- row.names(wide_df0) 
row.names(wide_df0) <- NULL

# 查看结果
wide_df0

简洁替代方案 #

利用dplyr确保分组唯一性，结合R基础包的reshape函数直接转换宽表，无需数值映射来回转换：

library(dplyr)

# 确保每个(SEZ, Class)组合仅保留一个Val（用户数据已满足，此步提升鲁棒性）
df_clean <- df1 %>%
  group_by(SEZ, Class) %>%
  summarise(Val = first(Val), .groups = "drop")

# 使用基础包reshape转换为宽表，并重命名列
wide_df <- reshape(df_clean, 
                   idvar = "SEZ",  # 行标识列
                   timevar = "Class",  # 转为列的变量
                   direction = "wide",  # 宽表方向
                   v.names = "Val") %>%  # 要展开的数值列
  # 重命名列，去掉前缀"Val."并添加"col_"前缀
  rename_with(~paste0("col_", sub("Val\\.", "", .))) %>%
  # 可选：将NA替换为指定文本（如"Unknown"）
  mutate(across(starts_with("col_"), ~replace_na(., "Unknown")))

# 查看结果
wide_df

方案优势 #

• 无需手动构建数值映射与反向映射，代码更简洁易读
• 直接基于文本值操作，避免数值转换可能带来的错误
• 结合dplyr的分组处理保证数据鲁棒性，同时仅使用允许的工具集

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