1

我有两个矩阵

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

4

3 回答 3

6

常规的逻辑和算术运算符对 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
于 2012-08-20T18:29:15.803 回答
5

如果这是真的,我不确定你想要哪个元素,所以我假设总和:

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].

于 2012-08-20T17:44:43.297 回答
1

在矩阵上使用简单的比较运算符。

重新创建数据,使用:

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
于 2012-08-20T17:45:04.607 回答