在R中，比循环或应用更快的方式

Question

对于这两个矩阵，我想找到矩阵X和Q行的乘积，并应用ifelse函数来查看产品是否大于零。

n1=1000, m=10000
X=cbind(rnorm(n1),rbinom(n1))
Q=matrix(rnorm(2*m), ncol=2)

为此，我尝试了循环并在下面应用函数。

D=10000
ind_beta=matrix(0,n1,D)
for (l in 1:D){
ind[,l]=as.vector(ifelse(X%*%Q[l,]>=0,1,0))   
}

和

ind=apply(Q,1,function(x){ifelse(X%*%Q>=0,1,0)})

两个代码都给出相同的结果，但实际上非常耗时。 有没有办法让这个快？提前谢谢。

Answer 1

怎么样：

制作数据（可重复）：

set.seed(101)
n1=1000; m=10000
X=cbind(rnorm(n1),rbinom(n1,size=1,prob=0.6))
Q=matrix(rnorm(2*m), ncol=2)

你的方式大约需要2.5秒：

system.time(ind <- apply(Q,1,function(x){ifelse(X%*%x>=0,1,0)}))

大约需要0.3秒：

system.time({
    XQ <- X %*% t(Q)
    ind2 <- matrix(as.numeric(XQ>=0),nrow(XQ))
})

结果匹配：

all.equal(ind,ind2) ## TRUE