Question

给定一个矩阵，我想找到每个元素的邻居之和（所以结果是一个矩阵）。如果存在（不考虑对角线元素），则邻居是给定元素上方，下方和旁边的值。

示例：

> z = matrix(1:9, 3, 3, byrow=T)
> z
     [,1] [,2] [,3]
[1,]    1    2    3
[2,]    4    5    6
[3,]    7    8    9

预期结果是：

> result
     [,1] [,2] [,3]
[1,]    6    9    8
[2,]   13   20   17
[3,]   12   21   14

在不使用循环的情况下，我可以在R中执行此操作的最简单方法是什么？

Answer 1

一种方法是在每一侧与邻居制作矩阵并将它们加在一起。

rbind(z[-1,],0) + rbind(0,z[-nrow(z),]) + cbind(z[,-1],0) + cbind(0,z[,-ncol(z)])
##      [,1] [,2] [,3]
## [1,]    6    9    8
## [2,]   13   20   17
## [3,]   12   21   14

Answer 2

这是一种方法。我不会声称它是最简单的，但它肯定会避免循环。

m <- matrix(1:16, 4, 4, byrow=TRUE)

result <- array(sapply(seq_along(m), function(i) {
  ind <- which(col(m) == col(m)[i] & abs(row(m)[i] - row(m)) == 1 | 
                 row(m) == row(m)[i] & abs(col(m)[i] - col(m)) == 1)
  sum(m[ind])
}), dim(m))


result

     [,1] [,2] [,3] [,4]
[1,]    7   10   13   11
[2,]   16   24   28   23
[3,]   28   40   44   35
[4,]   23   38   41   27

对于矩阵中的每个元素，找到其所有邻居的总和

2 个答案: