创建包含来自四个列表的所有可能值组合的数据表

Question

我有以下四个列表。

varnames <- list("beefpork", "breakfast", "breakfast_yn", "diet_soda", "food_label", "fruit_and_veggie", "fruit_juice", "fruits", "milk",                      "min_foods","regular_soda", "ssb", "total_fruit", "vegetables",                      "asthma", "bmiclass3", "bmiclass4","bmiclass5", "dental_absence",                     "dental_appt", "diabetes", "food_allergies", "sore_teeth", "trying_weight",                     "count_pa60days", "count_vigpa20days", "gaming_bedroom", "other_organized_pa", "pa30outdoor","paguidelines", "pc_time", "school_transport", "sport_teams", "tv_bedroom", "tv_time_char", "video_games_char")
grades <- list("2", "4", "8", "11")
groups <- list("none", "ethnic", "bordercounty")
regions <- list("state", "hsr")

以及以下函数，它返回一个整数：

all_empty = function(outcome, groupvar, gradevar, regionvar){
  #How many observations?

  if (groupvar == "none") 
    fmla <- as.formula(paste0("~", outcome))
  else 
    fmla <- as.formula(paste0("~", outcome, "+", groupvar))

  if (regionvar == "hsr")
    mydata = span_phrwts
  else if (regionvar == "state" & groupvar %in% c("none", "ethnic"))
    mydata = span_statewts
  else if (regionvar == "state" & groupvar == "bordercounty")
    mydata = span_borderwts
  else mydata = span_statewts

  myrow = svytable(fmla, subset(mydata, grade==gradevar)) %>% nrow()
  return(myrow)
}

我正在尝试编写一个代码，该代码将对列表中所有 864 种可能的值组合运行该函数，并创建一个具有 864 行和 5 列的数据表。

我希望决赛桌看起来像这样，但没有成功：

Variable          Grade          Group           Region     Obs
beefpork          2              none            state      5
beefpork          4              none            state      5
beefpork          8              none            state      3
beefpork          11             none            state      0

这是我尝试运行它，但我无法正确计算rownum。

output_all <- matrix(ncol = 5, nrow = length(varnames)*length(grades)*length(groups)*length(regions))
for(l in 1:length(regions)) {
  for (k in 1:length(grades)) {
    for(j in 1:length(groups)) {
      for(i in 1:length(varnames)){
        rownum = i + ((length(groups)*length(grades)*length(regions)) - 1)
        output_all[rownum, 1] = varnames[[i]]
        output_all[rownum, 2] = groups[[j]]
        output_all[rownum, 3] = grades[[k]]
        output_all[rownum, 4] = regions[[l]]
        output_all[rownum, 5] = all_empty(varnames[[i]], groups[[j]], grades [[k]], regions[[l]])

      }
    }
  } 
}
output_all %>% as_data_frame() %>% View()

任何帮助/建议将不胜感激！

Answer 1

使用您可以创建交叉连接的data.table功能。CJ然后我们添加一个行号（Idx）来执行函数的逐行调用。我们最终删除了 Idx 列

library(data.table)
dt <- CJ(varnames=varnames,grades=grades,groups=groups,regions=regions)
dt[,Idx:=.I]
dt[,by=Idx, Obs:=all_empty(outcome, groupvar, gradevar, regionvar)]
dt[,Idx:=NULL]

Answer 2

If it's ok to use vectors and not lists, tidyr::crossing seems like a straightforward approach.

varnames <- c("beefpork", "breakfast", "breakfast_yn", "diet_soda", "food_label", "fruit_and_veggie", "fruit_juice", "fruits", "milk",                      "min_foods","regular_soda", "ssb", "total_fruit", "vegetables",                      "asthma", "bmiclass3", "bmiclass4","bmiclass5", "dental_absence",                     "dental_appt", "diabetes", "food_allergies", "sore_teeth", "trying_weight",                     "count_pa60days", "count_vigpa20days", "gaming_bedroom", "other_organized_pa", "pa30outdoor","paguidelines", "pc_time", "school_transport", "sport_teams", "tv_bedroom", "tv_time_char", "video_games_char")
grades <- c("2", "4", "8", "11")
groups <- c("none", "ethnic", "bordercounty")
regions <- c("state", "hsr")


tidyr::crossing(varnames, grades, groups, regions)


# A tibble: 864 x 4
   varnames grades groups       regions
   <chr>    <chr>  <chr>        <chr>  
 1 asthma   11     bordercounty hsr    
 2 asthma   11     bordercounty state  
 3 asthma   11     ethnic       hsr    
 4 asthma   11     ethnic       state  
 5 asthma   11     none         hsr    
 6 asthma   11     none         state  
 7 asthma   2      bordercounty hsr    
 8 asthma   2      bordercounty state  
 9 asthma   2      ethnic       hsr    
10 asthma   2      ethnic       state