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

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)))
}