我有两个矩阵
A
2 1 0
0 1 6
1 0 0
B
1 1 3
4 2 8
2 0 1
我想找出比较满足以下条件的 A 和 B 的元素:
if A[i,j] =0 and B[i,j]>0
所以我的结果矩阵应该是
C
0 0 3
4 0 0
0 0 1
除了 a 之外,什么是正确的方法来做到这一点for loop
?
常规的逻辑和算术运算符对 R 中的矩阵按元素进行操作。只有 the%*%
和 kronecker 运算符进行矩阵乘法。此外,您可以使用“[”和“[<-”函数索引矩阵:
> C <- B
> C[!( A==0 & B >0) ] <- 0
> C
[,1] [,2] [,3]
[1,] 0 0 3
[2,] 4 0 0
[3,] 0 0 1
如果这是真的,我不确定你想要哪个元素,所以我假设总和:
A <- matrix(c(2, 1, 0, 0, 1, 6, 1, 0, 0),3,3,byrow=TRUE)
B <- matrix(c(1, 1, 3, 4, 2, 8, 2, 0, 1),3,3,byrow=TRUE)
C <- ifelse(A==0 & B>0, A+B, 0)
[,1] [,2] [,3]
[1,] 0 0 3
[2,] 4 0 0
[3,] 0 0 1
但更清晰的可能是使用which
:
which(A==0 & B>0, arr.ind=TRUE)
row col
[1,] 2 1
[2,] 1 3
[3,] 3 3
请注意,索引矩阵的正确方法是A[i,j]
代替A[i][j]
.
在矩阵上使用简单的比较运算符。
重新创建数据,使用:
A <- as.matrix(read.table(text="
2 1 0
0 1 6
1 0 0"))
B <- as.matrix(read.table(text="
1 1 3
4 2 8
2 0 1"))
执行比较:
A==0 & B>0
V1 V2 V3
[1,] FALSE FALSE TRUE
[2,] TRUE FALSE FALSE
[3,] FALSE FALSE TRUE
然后使用ifelse
:
ifelse(A==0 & B>0, B, 0)
V1 V2 V3
[1,] 0 0 3
[2,] 4 0 0
[3,] 0 0 1