R: Verwenden von Mapply, um Schleife zu vermeiden

Question

Ich habe folgenden Code, den ich mit for-Schleife ausführen, um ein 3D-Array "newarr" zu generieren.

n1<-c(1,2,3,4,5) 
n2<-c(3,4,5,6,7) 
n3<- c(4,5,6,7,8) 

afun <- function(y,p,q,r){ 
       calc=1/(1+(((y-p)/q)^(2*r))) 
       return(calc) 
     } 

newarr<- array(dim = c(4,5,5)) 
Amat<-matrix(data=c(1:10,NA,NA,NA,NA,15:20),nrow = 4,ncol = 5) 
Bmat<-matrix(data=c(1:6,NA,NA,NA,NA,11:20),nrow = 4,ncol = 5) 
Qmat<- +(!is.na(Amat) & !is.na(Bmat)) 

for(i in 1:5){ 
    res<-afun(Amat,n1[i],n2[i],n3[i]) 
    newarr[,,i]<- res 
} 

Ich möchte Mapply verwenden (oder eine Funktion anwenden) statt for Schleife. arr2 < - array ((mapply (funktion (x, y, n1, n2, n3) if (x == 1) afun (y, n1, n2, n3) sonst 0, Qmat, Amat, n1, n2, n3)), C (4,5,5))

Der folgende Code scheint aber ohne Bedingung zu helfen, wie weiter beschrieben:
newarr < - array (mapply (afun, n1, n2, n3, MoreArgs = list (y = Amat)), c (4,5,5))

Ich möchte eine Bedingung mit Qmat (eine 4x5-Matrix mit 0 und 1), so dass, wenn '0' beobachtet wird, sollte keine Operation enthalten durchgeführt werden und '0'-Wert zur Füllmatrix zurückgeben. Führen Sie für 1s in Qmat die Funktion 'afun' aus, und geben Sie den Wert zu "newarr" zurück, um eine 3D-Matrix zu bilden.

Danke für die Hilfe !!

Answer 1

Dies wird Ihnen das gewünschte Ergebnis:

newarr <- array(mapply(afun, n1,n2,n3, MoreArgs = list(y=Amat)), c(4,5,5)) 

dies ist die implizite SIMPLIFY mit = TRUE von mapply(). Mit SIMPLIFY = TRUE wird jedes Ergebnis auf einen Vektor reduziert. (Eine Matrix ist ein Vektor mit einer Dimension Attribut)
Eine andere Variante ist

array(unlist(mapply(afun, n1,n2,n3, MoreArgs = list(y=Amat), SIMPLIFY = FALSE)), c(4,5,5)) 

Hier finden Sie eine Liste von Matrizen als Ergebnis erhalten von mapply(). Sie müssen die Registrierung aufheben.