R语言矩阵操作：修改值为1的元素邻接值并处理边界异常问题

1. 判断条件错误：需求是筛选值为1的元素，代码中写的是y[row,col] == 2，逻辑完全不匹配
2. 无边界校验：直接对行、列索引做±1操作，当元素处于矩阵边缘/角落时，会生成小于1或者超过矩阵行列数的非法索引，触发越界报错
3. 遍历逻辑缺陷：直接在遍历过程中修改原矩阵，后续遍历到被修改过的元素时会误判为初始1元素，重复处理邻域

# 构造初始测试矩阵，和你给出的示例一致
y <- matrix(c(
  1,0,0,1,0,
  1,0,0,0,0,
  0,0,0,0,0,
  0,0,0,1,0,
  0,0,0,0,0
), nrow = 5, byrow = TRUE)

# 自定义参数：邻域元素取1的概率，0.5就是0和1各50%概率，可自行调整
prob_1 <- 0.5

# 第一步：提前提取所有初始值为1的元素坐标，避免后续修改矩阵影响判断逻辑
one_points <- which(y == 1, arr.ind = TRUE)
n_row <- nrow(y)
n_col <- ncol(y)

# 第二步：遍历每个初始为1的坐标，处理其8邻域
for (i in 1:nrow(one_points)) {
  r <- one_points[i, "row"]
  c <- one_points[i, "col"]
  
  # 生成8邻域的行列偏移组合
  offset_r <- c(-1, 0, 1)
  offset_c <- c(-1, 0, 1)
  neighbors <- expand.grid(r + offset_r, c + offset_c)
  colnames(neighbors) <- c("r", "c")
  # 排除当前中心点本身，只保留周围8个邻域
  neighbors <- neighbors[!(neighbors$r == r & neighbors$c == c), ]
  
  # 边界校验：只保留落在矩阵合法范围内的坐标，彻底避免越界
  valid_neighbors <- neighbors[
    neighbors$r >= 1 & neighbors$r <= n_row &
    neighbors$c >= 1 & neighbors$c <= n_col,
  ]
  
  # 给合法邻域按设定概率赋值0/1
  for (j in 1:nrow(valid_neighbors)) {
    nr <- valid_neighbors$r[j]
    nc <- valid_neighbors$c[j]
    y[nr, nc] <- sample(c(0,1), size = 1, prob = c(1-prob_1, prob_1))
  }
}

# 输出处理后的矩阵
print(y)

如果你需要处理其他尺寸的矩阵，只需要替换初始矩阵y的定义即可，边界适配逻辑不需要修改；调整prob_1参数即可修改邻域位置生成1的概率。

内容的提问来源于stack exchange，提问作者2000rs25