如何使用带有函数和输出的嵌套 for 循环答案

【问题标题】：How use nested for loops with a function and an output如何使用带有函数和输出的嵌套 for 循环
【发布时间】：2021-07-01 23:31:37
【问题描述】：

我在应用具有嵌套 for 循环的函数时遇到问题，因此请按个人和每月吐出数据值。目前我可以使用 for 循环应用该函数，以便每月获取数据。

我使用的数据集和函数非常大，但我在下面创建了一些示例数据和示例函数。

structure(list(code = c("a", "a", "a", "a", "a", "a", "b", "b", 
"b", "b", "b", "b", "c", "c", "c", "c", "c", "c"), datetime = c("16/04/2016", 
"17/04/2016", "25/05/2016", "26/05/2016", "01/06/2016", "02/06/2016", 
"16/05/2016", "17/05/2016", "25/06/2016", "26/06/2016", "01/07/2016", 
"02/07/2016", "16/06/2016", "17/06/2016", "25/07/2016", "26/07/2016", 
"01/08/2016", "02/08/2016"), score = c(17L, 16L, 12L, 16L, 14L, 
2L, 1L, 10L, 13L, 12L, 0L, 7L, 17L, 8L, 15L, 20L, 0L, 4L), monthyear = c("2016/04", 
"2016/04", "2016/05", "2016/05", "2016/06", "2016/06", "2016/05", 
"2016/05", "2016/06", "2016/06", "2016/07", "2016/07", "2016/06", 
"2016/06", "2016/07", "2016/07", "2016/08", "2016/08")), class = "data.frame", row.names = c(NA, 
-18L))

month_list <- strftime(seq(as.Date("2016/04/01"), as.Date("2016/08/31"), by = "month"),format="%Y/%m")

test_func <- function(dat) {
 metrics <- dat %>% summarize(
    mean = mean(score, na.rm = TRUE), 
    sd = sd(score, na.rm = TRUE)) 
  metrics$code <- rep(first(dat$code), nrow(metrics))
  metrics$monthyear <- rep(first(dat$monthyear), nrow(metrics))
  return(metrics)
}


my_datalist = list()

for (i in month_list) {
  # define outputs of function
  my_datalist[[i]]  <- testdat %>%
    filter(monthyear== i) %>% 
    test_func
  # add outputs to empty datalist
}

my_metric_data = do.call(rbind, my_datalist)

#turn into dataframe
my_metric_data = do.call(rbind, my_datalist)

这会返回一行数据，我的月份列表中的每个月都有一个数据。我现在需要每月将此函数（test_func）应用于数据集中的每个人。所以我想我会构建一个嵌套的 for 循环，在其中我每月过滤数据，创建该月的个人（代码）列表。然后将该函数应用于该列表。

my_datalist = list()

for (i in month_list) {
  dat <-  df %>%
    filter(monthyear== i)
  
  code_list  <- as.character(unique(dat$code))
  
  for (j in code_list){
    my_datalist[[j]] <- dat %>% 
      filter(code == j) %>% 
      test_func
  }
}

my_metric_data <- do.call(rbind, my_datalist)

但是，当我检查输出时，它看起来只是将函数应用于第一个代码，而不是按代码每月返回数据。但我不确定它为什么这样做。我认为我需要潜在地使另一个空列表来填充，然后添加到第一个列表中，但到目前为止我的尝试还没有奏效。

【问题讨论】：

什么是net_func？
啊，抱歉，它只是一个函数，它返回一些填充列表的数据值
除非完整，否则我们仍然无法运行代码
我已经用一些虚拟数据和虚拟函数更新了问题，以尝试显示我所追求的，

标签： r function for-loop nested-for-loop

【解决方案1】：

我们没有任何数据可以运行或验证解决方案，但您可以试试这种split + lapply 方法。

result <- do.call(rbind, lapply(split(GRS_filt, 
                  list(GRS_filt$monthyear, GRS_filt$code)), net_func))

【讨论】：

【解决方案2】：

我的一位同事帮我解决了这个问题，所以我想我会发布答案。

修复它的最简单方法可能是在运行循环之前设置一个索引计数器变量；

idx_cnt <- 1

然后在您的内部循环（ j 之一）中，使用它来索引结果列表，然后添加 1 以便下一个结果进入下一个插槽。生成的代码如下所示。

datalist = list()
idx <- 1

for (i in month_list) {
  dat <-  dat %>%
    filter(monthyear== i)
  
  code_obs <- dat %>% 
    group_by(code) %>% 
    summarise(n = n()) %>% 
    filter(n >=20) %>% 
    ungroup()
  
  code_list  <- as.character(unique(code_obs$code))
  
  for (j in code_list){
    datalist [[idx]] <- dat %>% 
      filter(code == j) %>% 
      nodeMetrics_func
    idx <- idx + 1
  }
}

【讨论】：