如何高效修改矩阵中值为1的元素的4个相邻单元格值

嘿，我完全懂你不想写冗长循环的痛点！在R里处理矩阵这类问题，向量化操作是最优解——既能简化代码，又能大幅提升效率。下面一步步帮你实现需求：

1. 先还原你的矩阵 #

先把你的矩阵定义明确下来，方便后续操作：

Y <- matrix(c(1,0,0,0,0,0,1,0,1,0,1,0,0,0,1,1), byrow=TRUE, nrow=4)

这个矩阵的初始状态是：

[,1] [,2] [,3] [,4]
[1,]    1    0    0    0
[2,]    0    0    1    0
[3,]    1    0    1    0
[4,]    0    0    1    1

2. 获取所有值为1的位置 #

用which()函数配合arr.ind=TRUE，直接得到所有1的行列索引矩阵，比手动找位置高效太多：

ones_pos <- which(Y == 1, arr.ind = TRUE)

输出的索引矩阵是：

row col
[1,]   1   1
[2,]   3   1
[3,]   2   3
[4,]   3   3
[5,]   4   3
[6,]   4   4

3. 定义四个相邻方向的偏移量 #

把上、下、左、右四个方向的行列偏移量做成一个矩阵，这样可以批量生成所有相邻位置：

# 每行代表一个方向：(行偏移, 列偏移)
directions <- matrix(c(-1, 0,  # 上
                       1, 0,   # 下
                       0, -1,  # 左
                       0, 1),  # 右
                     ncol=2, byrow=TRUE)

4. 生成所有相邻位置并过滤有效范围 #

通过重复索引的方式，把每个1的位置和四个方向做组合，得到所有可能的相邻位置，再过滤掉超出矩阵边界的无效位置：

# 生成所有相邻位置（每个1对应4个方向）
adjacent_pos <- ones_pos[rep(1:nrow(ones_pos), each=4), ] + directions[rep(1:4, nrow(ones_pos)), ]

# 过滤掉超出矩阵范围的位置（行≥1、行≤总行数；列≥1、列≤总列数）
valid_pos <- adjacent_pos[
  adjacent_pos[,1] >= 1 & adjacent_pos[,1] <= nrow(Y) &
  adjacent_pos[,2] >= 1 & adjacent_pos[,2] <= ncol(Y), 
]

5. 批量修改相邻单元格的值 #

现在直接用索引批量修改就行！比如你要改成数字2：

Y[valid_pos] <- 2

修改后的矩阵：

[,1] [,2] [,3] [,4]
[1,]    1    2    0    0
[2,]    2    2    1    2
[3,]    1    2    1    2
[4,]    0    2    1    1

如果要改成句号.，记得先把矩阵转成字符型（因为原矩阵是数值型，不能直接存储字符）：

# 转成字符矩阵
Y_char <- as.character(Y)
# 修改有效位置为.
Y_char[valid_pos] <- "."
# 转回矩阵格式（可选）
Y_char <- matrix(Y_char, nrow=nrow(Y))

修改后的字符矩阵：

[,1] [,2] [,3] [,4]
[1,] "1"  "."  "0"  "0" 
[2,] "."  "."  "1"  "." 
[3,] "1"  "."  "1"  "." 
[4,] "0"  "."  "1"  "1" 

额外优化：避免重复修改同一个位置 #

如果某个单元格是多个1的相邻，会被多次修改（不过结果不影响），如果想去掉重复的位置，可以用unique()：

valid_pos_unique <- unique(valid_pos)
Y[valid_pos_unique] <- 2

这种向量化的方式完全不需要逐行循环，代码简洁且效率更高——尤其是当矩阵很大的时候，优势会特别明显！

内容的提问来源于stack exchange，提问作者Tony Hellmuth