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

Question

我用这个例子来解释如何用 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))

Answer 1

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

variable1 和 variable2 随机设置以说明myFunction中 myData（x 和 y）中的数据。

for 循环正在探索国家和年份的组合……这两个代码（for 循环和 lappy）不会产生完全相同的结果。 lapply 将生成一个列表，其中包含myFunction的结果。 for 循环将生成一个数据帧。

Answer 2

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

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