Używanie "apply" do zastosowania funkcji do macierzy, w której parametry są specyficzne dla kolumny

Question

Próbuję uniknąć używania pętli, używając apply do zastosowania funkcji zdefiniowanej przez użytkownika do macierzy. Problem polega na tym, że istnieją dodatkowe parametry, których używa moja funkcja i które różnią się dla każdej kolumny macierzy. Poniżej znajduje się przykład zabawki.

że mam następującą funkcję:

foo <- function(x, a, b, c) return((a*x + b)^c) 

i chcę, aby zastosować go do matrycy bar używając różnych wartości a, b i c dla każdej kolumny.

bar <- matrix(1:15, ncol = 3) 
a <- 4:6 
b <- 3:1 
c <- 1:3 

W tym przypadku, w pierwszej kolumnie, po czym a=4bar, b=3 i c=1. Próbowałem tego, ale jest to oczywiście niepoprawne, ponieważ każda kolumna używa kolejno wszystkich parametrów przed zawinięciem do pierwszego parametru ponownie. Jakieś sugestie?

Answer 1

Mogliśmy split „Pasek” w kolumnie „” (col(bar)) i mapply można zastosować „foo” do odpowiadającej „a”, „b”, wartości „C” do każdej kolumny „bar”

mapply(foo, split(bar, col(bar)), a, b, c) 

Albo bez użycia apply

ind <- col(bar) 
(a[ind]*bar +b[ind])^c[ind] 

Answer 2

można umieścić swoje parametry do wektora:

newbar <- rbind(a,b,c,bar) 
newfoo <- function(z){x <- z[-(1:3)]; (z[1]*x+z[2])^z[3]} 
apply(newbar,2,newfoo) 

co daje

[,1] [,2] [,3] 
    7 1024 300763 
    11 1369 389017 
    15 1764 493039 
    19 2209 614125 
    23 2704 753571 

Answer 3

Można użyć sweep; zwykle jest to dla odjęcie średniej z każdej kolumny, ale można przejść w indeksie zamiast dostać indeksowanie równoległego poprzek a, b i c:

> sweep(bar, 2, seq_along(a), function(x,i) foo(x, a[i], b[i], c[i]), FALSE) 
    [,1] [,2] [,3] 
[1,] 7 1024 300763 
[2,] 11 1369 389017 
[3,] 15 1764 493039 
[4,] 19 2209 614125 
[5,] 23 2704 753571 

Answer 4

nie widzę dlaczego męczyć się z funkcji w ogóle :

> a <- matrix(4:6,nrow = 5,ncol = 3,byrow = TRUE) 
> b <- matrix(3:1,nrow = 5,ncol = 3,byrow = TRUE) 
> c <- matrix(1:3,nrow = 5,ncol = 3,byrow = TRUE) 
> (a*bar + b)^c 
    [,1] [,2] [,3] 
[1,] 7 1024 300763 
[2,] 11 1369 389017 
[3,] 15 1764 493039 
[4,] 19 2209 614125 
[5,] 23 2704 753571 

Answer 5

myślę, że można po prostu przenieść swój matrycy i używać wektoryzacji:

t(foo(t(bar),a,b,c)) 

to powinno działać na każdym wektorowy foo.