如何定义一定数量的嵌套 for 循环（基于 R Shiny 应用程序中的输入长度）？

Question

这里是上下文：我在一个 R 闪亮的 Web 应用程序上工作。用户上传一个数据框。然后，他使用 selectInput 选择一定 n 个列。所选的列数可以从 1 到 6 不等。

基于此列数，我想自动生成适当数量的嵌套 for 循环。那时，我使用 if() 条件来测试每个可能选择的列数。 我想传递所选每一列的每个唯一值。这使我的代码很长：

my_columns = input$colnames #The user selects column names

if(length(mycolumns) == 1){
 for(var1 in unique(mydataframe[,my_columns[1]])){
   ...
 }
}

if(length(mycolumns) == 2){
 for(var1 in unique(mydataframe[,my_columns[1]])){
    for(var2 in unique(mydataframe[,my_columns[2]])){
     ...
    }
  }
}

if(length(mycolumns) == 3){
 for(var1 in unique(mydataframe[,my_columns[1]])){
   for(var2 in unique(mydataframe[,my_columns[2]])){
     for(var3 in unique(mydataframe[,my_columns[3]])){
      ...
     }
   }
 }
}

等等……

有没有办法避免这种情况？

谢谢

Answer 1

如果我弄错了，请纠正我，但您似乎计算了一些需要涵盖所选列的所有可能值组合的东西。

在这种情况下，R 不需要嵌套的 for 循环

my_columns <- data.frame(
  "A" = c(1,2,3),
  "B" = c(11,12,13),
  "C" = c(21,22,23))

# find all unique values per column
list_uniques <- lapply(seq_along(my_columns),
                       function(x){unique(my_columns[[x]])}
                       )

# find out all possible combinations of the given values
# the output is a dataframe
all_combinations <- expand.grid(list_uniques)

# Now you can iterate over the frame and do something with them

# example rowsums
rowSums(all_combinations) # vectorized functions like this are faster

# example custom function
apply(all_combinations,
              MARGIN = 1, # iterate rowwise
              # you can now use your own function
              # the input i is a row as a named vector
              FUN = function(i){paste(i,collapse = " and ")})
# This function will output:
# "1 and 11 and 21" "2 and 11 and 21" ....