在 R 中滚动 window 模式更新

Question

我想在矩阵上应用滚动 window 模式更新。假设我有一个零和一个矩阵，如下所示：

mat <- matrix(c(0,1,1,0,1,1,0,0,1,1,1,0,1,0,0,0),nrow=4)

0   1   1   1
1   1   1   0
1   0   1   0
0   0   0   0

我想一次扫描矩阵的每一行 2 个值，并根据这组规则创建一个标签矩阵：

0   0   - then label as A
0   1   - then label as B
1   0   - then label as B
1   1   - then label as C

上述矩阵的期望输出是：（出现 NA 值是因为第一个值之前没有值）

NA  B   C   C
NA  C   C   B
NA  B   B   B
NA  A   A   A

我正在尝试在 R 中实现它。有没有办法使用其中一种应用方法来做到这一点？或者是否有一种快速方法将此数据存储为 data.frame 并使用 dplyr 或 data.table 等包进行处理？目前，我将其作为 for 循环处理（当然，非常非常慢）。

Answer 1

这里的rolling操作作用于pairwise组合，所以你可以直接定义函数：

f <- function(x,y) LETTERS[1:3][x+y+1L]

然后将其应用于适当的对：

nc  <- ncol(m)
res <- matrix(,length(m)/nc,nc)

res[,-1] <- f(m[,-nc], m[,-1])

#      [,1] [,2] [,3] [,4]
# [1,] NA   "B"  "C"  "C" 
# [2,] NA   "C"  "C"  "B" 
# [3,] NA   "B"  "B"  "B" 
# [4,] NA   "A"  "A"  "A"

在 R 中滚动 window 模式更新

Rolling window pattern update in R

r

matrix

apply

dplyr