在R语言中将行列数不同的数据框列表转换为三维数组

问题背景 #

你有如下数据框和列表定义：

df1 <- data.frame(ID=1:3, test=2:4, category=1)   
df2 <- data.frame(ID=1:2, test=4:5, test1=2:3, category=2)
Mylist <- list(df1, df2)

直接使用array(unlist(Mylist))无法得到期望的三维数组结构，因为两个数据框的行列数不一致，无法直接指定统一的数组维度。

解决方案1：使用abind包（推荐） #

abind包专门用于合并不同维度的数组/矩阵，能直接按第三维度合并不同行列数的数据框（需先转为矩阵）。

# 首次使用需安装包
install.packages("abind")
library(abind)

# 将列表中的数据框转为矩阵
mat_list <- lapply(Mylist, as.matrix)

# 按第三维度合并
array1 <- abind(mat_list, along = 3)

# 查看结果
array1[,,1]
#>      ID test category
#> [1,]  1    2        1
#> [2,]  2    3        1
#> [3,]  3    4        1

array1[,,2]
#>      ID test test1 category
#> [1,]  1    4     2        2
#> [2,]  2    5     3        2

该方法会严格保留每个数据框的行列结构，生成的三维数组完全符合预期。

解决方案2：手动构造数组（无依赖） #

如果不想依赖第三方包，可以手动创建一个足够大的数组，用指定值（如NA）填充空缺后，逐个填入数据框内容：

# 获取每个数据框的行、列数
row_counts <- sapply(Mylist, nrow)
col_counts <- sapply(Mylist, ncol)

# 创建最大维度的空数组，用NA填充空缺
max_row <- max(row_counts)
max_col <- max(col_counts)
array1 <- array(NA, dim = c(max_row, max_col, length(Mylist)))

# 逐个填充数据
for(i in seq_along(Mylist)){
  array1[1:row_counts[i], 1:col_counts[i], i] <- as.matrix(Mylist[[i]])
}

# 查看结果
array1[,,1]
#>      [,1] [,2] [,3]
#> [1,]    1    2    1
#> [2,]    2    3    1
#> [3,]    3    4    1

array1[,,2]
#>      [,1] [,2] [,3] [,4]
#> [1,]    1    4    2    2
#> [2,]    2    5    3    2
#> [3,]   NA   NA   NA   NA

如果不需要NA，可以将array(NA, ...)中的NA替换为你需要的填充值（如0）。

为什么直接unlist不行？ #

unlist(Mylist)会把所有数据框的元素按列优先顺序扁平化成长向量，丢失了原数据框的行列结构信息，而两个数据框的行列数不一致，无法通过指定单一维度参数还原成你想要的三维数组。

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