Есть ли какая-нибудь функция перекрестного применения в R? - PullRequest
3 голосов
/ 05 октября 2011

Есть ли какая-нибудь функция в R вида:

crossApply(v1, v2, func)

, который имеет такую ​​же функциональность, как:

ret = c()
i = 1
for (e1 in v1) {
  for (e2 in v2) {
    ret[i] <- func(e1,e2)
    i <- i + 1
  }
}
return(ret)

Заранее спасибо.

Ответы [ 3 ]

6 голосов
/ 05 октября 2011

Я думаю, вы можете искать outer, что не совсем то, что делает ваш код, но оно близко. В частности, outer будет возвращать матрицу (то есть внешнее произведение) или каждую комбинацию элементов своих первых двух аргументов.

Возможно, вы захотите сохранить результат, а затем извлечь нижний треугольник как вектор. Может быть, что-то вроде этого:

rs <- outer(1:4,-(5:7),"+")
rs[lower.tri(rs,diag = TRUE)]
[1] -4 -3 -2 -1 -4 -3 -2 -4 -3
1 голос
/ 05 октября 2011

Пример

func <- function(x,y) {sqrt(x^2+y^2)}
v1 <- c(1,3,5)
v2 <- c(0,-4,-12)
ret <- outer(v1,v2,"func")

И тогда у вас есть

> ret
     [,1]     [,2]     [,3]
[1,]    1 4.123106 12.04159
[2,]    3 5.000000 12.36932
[3,]    5 6.403124 13.00000

или если вы хотите именно то, что будут делать ваши циклы for

> as.vector(t(ret))
[1]  1.000000  4.123106 12.041595  3.000000  5.000000 12.369317  5.000000
[8]  6.403124 13.000000
0 голосов
/ 05 октября 2011

Это довольно легко сделать с do.call и expand.grid:

x <- seq(0,10, length.out=10)
> y <- seq(-1,1, length.out=5)
> d1 <- expand.grid(x=x, y=y)  
> do.call("*", d1)
 [1]   0.0000000  -1.1111111  -2.2222222  -3.3333333  -4.4444444
 [6]  -5.5555556  -6.6666667  -7.7777778  -8.8888889 -10.0000000
[11]   0.0000000  -0.5555556  -1.1111111  -1.6666667  -2.2222222
[16]  -2.7777778  -3.3333333  -3.8888889  -4.4444444  -5.0000000
[21]   0.0000000   0.0000000   0.0000000   0.0000000   0.0000000
[26]   0.0000000   0.0000000   0.0000000   0.0000000   0.0000000
[31]   0.0000000   0.5555556   1.1111111   1.6666667   2.2222222
[36]   2.7777778   3.3333333   3.8888889   4.4444444   5.0000000
[41]   0.0000000   1.1111111   2.2222222   3.3333333   4.4444444
[46]   5.5555556   6.6666667   7.7777778   8.8888889  10.0000000
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...