两个矩阵的 R 中的 sumif 函数

Question

我有两个包含时间序列数据的矩阵

A
2 1 0
0 1 6
1 4 6

B 
1 1 3
4 2 8
2 5 1

我想创建一个比较 A 和 B 并满足以下条件的向量：

如果 B 所以我的结果矩阵应该是

C
3 (=2+1)
1
7 (=1+6) 

我尝试过以下方式，但我无法获得正确的结果

posneg_fun<-function(x,y)
{
  new<-sum(x[y<3])
  return(new)
}

out<-t(apply(x,1,FUN=posneg_fun,y))

有什么想法吗？

谢谢

Answer 1

怎么样：

A[B>=3] <- NA
rowSums(A,na.rm=TRUE)
## [1] 3 1 7

或@Roland 的建议（更短）：

rowSums(A*(B<3))

它利用了TRUE 在进行数值运算时被强制为 1 而FALSE 被强制为 0 的事实......