用 lapply() 替换嵌套的 for 循环答案

【问题标题】：replacing nested for loop with lapply()用 lapply() 替换嵌套的 for 循环
【发布时间】：2020-06-03 18:08:18
【问题描述】：

我发送了这个例子来解释如何用 lapply() 函数替换嵌套的 for 循环。但是我不完全理解嵌套 for 循环中发生了什么？

根据我的理解，for 循环为所有年份的每个国家/地区创建了两个新变量tempX 和tempY，但是在 for 循环中参数的最后一行发生了什么？

variable1 和 variable2 的用途是什么？

# Generate random data:
allCountries <- LETTERS[1:10]
allYears <- 1990:2012

myData <- expand.grid(allCountries, allYears)  # create a dataframe with all possible combinations
colnames(myData) <- c("Country", "Year")
myData$variable1 <- rnorm(nrow(myData))
myData$variable2 <- rnorm(nrow(myData))

# Silly function to perform
myFunction <- function(x, y){
  x * y - x / y
}

### Doubly-nested loop ###
myData$computedFigure <- NA  # Make an "empty" variable in my data.frame

for(ii in allCountries){
  for(jj in allYears){
    tempX <- myData[myData$Country == ii & myData$Year == jj, c("variable1")]
    tempY <- myData[myData$Country == ii & myData$Year == jj, c("variable2")]
    # "Save" results into appropriate location in my data.frame
    myData[myData$Country == ii & myData$Year == jj, c("computedFigure")] <- myFunction(tempX, tempY)
  }
}

### Simple lapply() approach ###
computedFigureList <- lapply(1:nrow(myData), function(x){
  tempX <- myData[x, c("variable1")]
  tempY <- myData[x, c("variable2")]
  # "Save" results into appropriate location in my data.frame
  myFunction(tempX, tempY)
})

myData$computedFigure2 <- unlist(computedFigureList)
with(myData, plot(computedFigure, computedFigure2))

【问题讨论】：

variable1 和 variable2 只是数据。可能是 GDP($) 和 MedianIncome($)。或者您可能对其进行数学运算的任何其他变量。它们只是组成的数据。
for 循环的最后一行在数据框中定位您要放置结果的位置myData[...,...]，然后将数学结果放在那里myFunction(tempX,tempY。
不要使用lapply 和匿名函数，你可以只使用myData$computedFigure <- mapply(myFunction, myData$variable1, myData$variable2)

标签： r function for-loop apply lapply

【解决方案1】：

在循环myData[myData$Country == ii & myData$Year == jj, c("computedFigure")] <- myFunction(tempX, tempY)的最后一行，函数myFunction被应用并记录在computedFigure列中。

variable1和variable2随机设置，说明myFunction中myData(x和y)中的数据。

for 循环正在探索国家和年份的组合... 两个代码（for loop 和 lappy）不会产生完全相同的结果。 lapply 将使用myFunction 的结果生成一个列表。 for 循环将生成一个数据帧。

【讨论】：

【解决方案2】：

其实你并不需要嵌套*apply函数，你实际上可以使用outer + diag来计算computedFigure，它可以达到与嵌套for循环相同的结果。

myData$computedFigure <- diag(with(myData,outer(variable1,variable2,myFunction)))

【讨论】：