如何在 r 中一次创建数据框（不仅仅是一个）

Question

这是无序组合的另一个问题，并将结果存储在 r 中的矩阵中。

现在我有一个如下的数据框

>head(plan)
  bal midway coro cab ljc ot
1   1      1    1   2   2  2
2   1      1    2   1   1  2
3   1      1    2   1   2  2
4   1      1    2   2   1  2
5   1      1    2   2   2  1
6   1      2    1   1   2  2

我想提取每行中等于 1 的元素，使用其列名并排列它们，以存储在新的数据框中，例如第一行的day_1_1 ：

> permutations(3, 3, v = names(plan)[which(plan[1,] == 1, arr.ind=T)[, "col"]])
     [,1]     [,2]     [,3]    
[1,] "bal"    "coro"   "midway"
[2,] "bal"    "midway" "coro"  
[3,] "coro"   "bal"    "midway"
[4,] "coro"   "midway" "bal"   
[5,] "midway" "bal"    "coro"  
[6,] "midway" "coro"   "bal"  

我的问题是我不知道如何在循环中创建名为day_1_i （ i匹配plan中的行号）的新数据框。 我试过了

for (i in 1:nrow(plan)) {
  paste0("day_1_", i) <- permutations(3, 3, v = names(plan)[which(plan[i,] == 1, arr.ind=T)[, "col"]])
}

但它不起作用。 我已经看到了一种可能的解决方案，使用assign from Using a loop to create multiple data frames in R ，但建议不要使用。 非常感谢指教！

Answer 1

您可以将其存储在数据框列表中

library(gtools)

list_df <- list()
for (i in 1:nrow(plan)) {
   list_df[[i]] <- data.frame(permutations(3, 3, 
           v = names(plan)[which(plan[i,] == 1, arr.ind=T)[, "col"]]))
}

然后如果您需要，您可以将其重命名为您的选择

list_df <- setNames(list_df, paste0("day_1_", 1:nrow(plan)))

list_df
#$day_1_1
#      X1     X2     X3
#1    bal   coro midway
#2    bal midway   coro
#3   coro    bal midway
#4   coro midway    bal
#5 midway    bal   coro
#6 midway   coro    bal

#$day_1_2
#   X1  X2  X3
#1 bal cab ljc
#2 bal ljc cab
#3 cab bal ljc
#4 cab ljc bal
#5 ljc bal cab
#6 ljc cab bal
#....
#....

因此，您现在可以按名称list_df[["day_1_1"]] 、 list_df[["day_1_2"]]等访问单个数据帧。

Answer 2

您可以使用split使用您喜欢的因子将 1 个数据框拆分为多个数据框的列表。 例如，下面将根据id列将df拆分为 5 个数据帧

df <- data.frame(id = 1:5, Val = rnorm(5))
split(df, df$id)

如果你想使用这也将工作rownames的data.frame，而不是和id列：

split(df, rownames(df))