关于apply函数的应用的小问题

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

cp是一个2*n的矩阵，treecut是一个大矩阵groupfinding(m,n)是一个输入参数为正整数的函数，返回值是一个整数（用在后面的程序里定位向量内的元素）：
groupfinding <- function(m,n){
    return(2205-max(which(treecut[m,]==treecut[n,])))
}


现在需要将cp的每一行的两个元素作为groupfinding的两个输入参数进行运行，我知道这可以用for循环来实现：
for(n in 1:(length(cp)/2)){
    for(n in 1:(length(cp)/2)){
        m<-cp[n,1];
        n<-cp[n,2];
        Haddingup<-Haddingup+HC$height[groupfinding2(m,n)];
    }


其中Haddingup的初始值是0
但是for循环太慢了，有没有什么方法可以用apply来代替for循环？

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏0 回帖

关键词：apply函数 apply appl App 小问题 apply函数

1. Haddingup <- 0
   
2. HaddingPlus <- function(i){
   
3.     Haddingup <<- Haddingup+HC$height[groupfinding(cp[i,1],cp[i,2])]
   
4. }
   
5. sapply(1:(length(cp)/2),HaddingPlus)

复制代码

不知道你代码里的groupfinding2是什么东西
"<<-"能在函数里给全局变量赋值

把你的groupfinding函数改成矢量形式，然后apply就可以了。可以看看下面的例子。不过我觉得速度应该差不多

1. mat <- matrix(rnorm(20000), ncol = 2)
   
2. compute2 <- function(vec) {
   
3.     return(vec[1]^2 + 2*vec[2] + vec[1]*vec[2])
   
4. }
   
5. out <- apply(mat, 1, compute2)
   
6. sum(out)

复制代码