замена вложенного для l oop на lapply () - PullRequest
Новая
       57

замена вложенного для l oop на lapply ()

0 голосов
/ 19 февраля 2020

Я отправил этот пример, чтобы объяснить, как заменить вложенную для l oop функцию lapply (). Однако я не до конца понимаю, что происходит во вложенном для l oop?

согласно моему пониманию, для l oop создает для каждой страны на все годы две новые переменные, называемые tempX и tempY, но что происходит в последней строке аргумента для for oop?

какова цель 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))

Ответы [ 2 ]

0 голосов
/ 19 февраля 2020

На самом деле вам не нужны вложенные *apply функции, вы можете использовать outer + diag для вычисления computedFigure, что может привести к тем же результатам, что и вы с помощью вложенных for l oop . 

myData$computedFigure <- diag(with(myData,outer(variable1,variable2,myFunction)))
0 голосов
/ 19 февраля 2020

В последней строке l oop myData[myData$Country == ii & myData$Year == jj, c("computedFigure")] <- myFunction(tempX, tempY) , функция myFunction применяется и записывается в столбец computedFigure.

variable1 и variable2 устанавливаются случайным образом для иллюстрации данных в myData (x и y) в myFunction.

Циклы for изучают комбинации в странах и годах ... Два кода (для l oop и lappy) не будут давать точно такой же результат , Lapply сгенерирует список только с результатом myFunction. Циклы for сгенерируют фрейм данных.

...