Warm tip: This article is reproduced from serverfault.com, please click

其他-通过在R中覆盖相同大小的布尔掩码和相同的输出尺寸来过滤矩阵

(其他 - Filter Matrix by overlaying boolean mask of same size with same output dimensions in R)

发布于 2020-12-04 10:38:22

给定此5x5邻接矩阵:

library(matrixcalc)
set.seed(1)
matrix(runif(25, 0, 10), nrow=5, ncol=5) %>% upper.triangle

得出此矩阵:

     [,1]     [,2]     [,3]     [,4]     [,5]
[1,] 2.655087 8.983897 2.059746 4.976992 9.347052
[2,] 0.000000 9.446753 1.765568 7.176185 2.121425
[3,] 0.000000 0.000000 6.870228 9.919061 6.516738
[4,] 0.000000 0.000000 0.000000 3.800352 1.255551
[5,] 0.000000 0.000000 0.000000 0.000000 2.672207

我想在此布尔过滤器掩码中过滤所有为TRUE的条目,同时保持大小,用NA或0填充FALSE条目。

mask <- matrix(rep(FALSE, 25), nrow=5, ncol=5)
mask[1,2] <- TRUE
mask[2,3] <- TRUE
mask[1,5] <- TRUE 
mask[3,5] <- TRUE

产生此掩码:

     [,1]  [,2]  [,3]  [,4]  [,5]
[1,] FALSE  TRUE FALSE FALSE  TRUE
[2,] FALSE FALSE  TRUE FALSE FALSE
[3,] FALSE FALSE FALSE FALSE TRUE
[4,] FALSE FALSE FALSE FALSE FALSE
[5,] FALSE FALSE FALSE FALSE FALSE

要输出由遮罩过滤的相同大小的矩阵,请执行以下操作:

    [,1]     [,2]     [,3] [,4]     [,5]
[1,]    0 8.983897 0.000000    0 9.347052
[2,]    0 0.000000 1.765568    0 0.000000
[3,]    0 0.000000 0.000000    0 6.516738
[4,]    0 0.000000 0.000000    0 0.000000
[5,]    0 0.000000 0.000000    0 0.000000
Questioner
user13607508
Viewed
0
Ronak Shah 2020-12-04 18:43:23

如果调用了矩阵,则mat可以执行以下操作:

mat[!mask] <- 0