Question

有没有人知道更有效的方法来执行以下操作？我有两个矩阵，一个是整数值，另一个是数字。我还有一个包含整数向量的列表。我有一个循环，当列表中的整数值等于整数矩阵时，它将数值矩阵的值设置为NA。然后我从结果矩阵中得到行积和求和。有没有办法避免每一步创建数字矩阵的副本？或者也许是另一种方法？感谢

mat1<-matrix(rpois(20*300000,6),20,300000)
mat2<-matrix(runif(20*300000),20,300000)
list1<-list(c(1,2,3),c(4,6),c(8,9,10,11))
results<-vector('numeric',length(list1))

start.time=Sys.time()
for(i in 1:length(list1)){
    copy<-mat2
    copy[mat1 %in% list1[[i]]]=NA
    results[i]=sum(apply(copy,2,prod,na.rm=T))
}
print(Sys.time()-start.time)
#

Answer 1

用apply替换你的colSums（假设你的数字是正数，否则你需要更多的摆弄）让我接近2倍的速度提升：

for(i in 1:length(list1)){
    copy<-mat2
    copy[mat1 %in% list1[[i]]]=NA
    results[i]=sum(exp(colSums(log(copy), na.rm = T)))
}

R循环替换矩阵中的多个值

1 个答案: