用 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)))