基于R中的多个列表/矩阵应用/循环定制函数

时间:2018-06-12 12:10:31

标签: r loops matrix apply custom-function

我有四个大型列表,我使用此函数转换为矩阵:

IndexOutOfBoundsException

我已将列表取消列表,将其转换为矩阵形式(不确定是否需要,但array(as.matrix(unlist(mat2)), dim=c(3, 80, 100)) 作为矩阵显着缩小)。如果使用列表更容易,请说出来。矩阵/列表有132个元素,但为简单起见,我们假设这里只有100个元素。

object.size

mat1

num [1:80, 1:100] 0.00669 0.00603 0.00895 0.00771 0.01533 ...

mat2

num [1:80, 1:80, 1:100] 0.0033 -0.00474 -0.00261 0.00587 -0.00599 ...

mat3

num [1:3, 1:80, 1:100] 0.0159 -0.0131 -0.0131 -0.0656 0.0554 ...

mat4

这是我的功能:

 num [1:10, 1:3] 0.00545 0.01043 0.00563 0.00431 0.00464 ...

而不是跑步:

customfunction <- function
(
  mat1,         # num [1:80, 1:100]
  mat2,         # num [1:80, 1:80, 1:100]
  mat3=NULL,    # num [1:3, 1:80, 1:100]
  mat4=NULL,    # num [1:10, 1:3]
  a=0.025       # scalar
)
   {
  omega = diag(c(1,diag(a * mat3 %*% mat2 %*% t(mat3))))[-1,-1]

  temp = solve(t(mat3) %*% solve(omega) %*% mat3 + solve(a * mat2))

  out = temp %*% (solve(a * mat2) %*% mat1 + t(mat3) %*% solve(omega) %*% mat4)

  return(out)
}

customfunction(mat1[1,], mat2[,,1], mat3[,,1], mat4[1,])
customfunction(mat1[2,], mat2[,,2], mat3[,,2], mat4[1,])
customfunction(mat1[3,], mat2[,,3], mat3[,,3], mat4[1,])
...
customfunction(mat1[61,], mat2[,,61], mat3[,,61], mat4[2,])
customfunction(mat1[61,], mat2[,,61], mat3[,,61], mat4[2,])
etc...

请注意customfunction(mat1[1,], mat2[[1]], mat3[[1]], mat4[1,]) customfunction(mat1[2,], mat2[[2]], mat3[[2]], mat4[1,]) customfunction(mat1[3,], mat2[[3]], mat3[[3]], mat4[1,]) ... customfunction(mat1[61,], mat2[[61]], mat3[[61]], mat4[2,]) customfunction(mat1[61,], mat2[[61]], mat3[[61]], mat4[2,]) 改变了第60次迭代,即我将其应用60次,保持mat4[i,]常量,然后应用下一次第60次迭代mat4[1,]。是否可以应用(或循环)它使每个元素/行运行函数100次,这样它将创建一个100x80矩阵?如果是这样,怎么样?

非常感谢!

1 个答案:

答案 0 :(得分:2)

这是sapply()的任务:

sapply(1:100, function(i) customfunction(mat1[i,], mat2[,,i], mat3[,,i], mat4[1+ (i-1)%/%60,]))

sapply(1:100, function(i) customfunction(mat1[i,], mat2[[i]], mat3[[i]], mat4[1+ (i-1)%/%60,]))