R语言如何对矩阵每列随机选取一个非零元素并将其减1

你可以直接通过逐列索引+随机采样的方式实现需求，完整代码如下：

# 构造初始矩阵
a <- c(1,2,0)
b <- c(0,5,0)
c <- c(8,9,2)
d <- c(5,2,0)
MyMatrix <- cbind(a,b,c,d)

# 逐列执行减1操作
for (col in seq_len(ncol(MyMatrix))) {
  # 提取当前列所有大于0的元素的行位置
  valid_pos <- which(MyMatrix[, col] > 0)
  # 随机抽取1个有效位置
  pick <- sample(valid_pos, 1)
  # 对应位置数值减1
  MyMatrix[pick, col] <- MyMatrix[pick, col] - 1
}

逻辑说明 #

• 用seq_len(ncol(MyMatrix))生成所有列的序号，逐列遍历处理
• which(MyMatrix[, col] > 0)会自动过滤掉列中的0值，只返回非零元素的行索引，保证后续采样不会选到0
• sample(valid_pos, 1)从有效位置中随机选1个，满足随机选取的要求
• 直接对选中的矩阵位置做减1赋值，修改原矩阵内容

兼容处理 #

如果你的矩阵可能存在全为0的列，可以加一行判断避免报错，修改后的核心逻辑如下：

for (col in seq_len(ncol(MyMatrix))) {
  valid_pos <- which(MyMatrix[, col] > 0)
  # 若当前列没有非零元素，直接跳过
  if (length(valid_pos) == 0) next
  pick <- sample(valid_pos, 1)
  MyMatrix[pick, col] <- MyMatrix[pick, col] - 1
}

运行代码后得到的结果和你给出的示例效果一致，每次运行因为随机采样的位置不同，结果会有差异，符合随机选取的要求。

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