如何在R语言中实现自定义矩阵右移函数的n次重复调用

你可以通过以下两种方式实现重复调用移位函数n次的需求：

方案1：基于你现有shift函数封装，直接重复调用n次 #

该方案完全复用你已写好的单步移位逻辑，仅需新增一层循环控制调用次数即可：

shift_n <- function(disc, n) {
  # 对移动次数取模，过滤无意义的重复循环（12列矩阵移动12次等于回到原位）
  n <- n %% ncol(disc)
  res <- disc
  for (i in seq_len(n)) {
    res <- shift(res)
  }
  return(res)
}

使用方法：直接传入待移位矩阵和移动次数即可，例如要右移3次执行shift_n(disc0, 3)，运行结果和你手动调用3次shift的结果完全一致。

方案2：更高效的通用版实现（无需依赖原有shift函数） #

直接通过列索引重排实现一步到位的移位，不用多次循环迭代，性能更高，且适配任意行列数的矩阵：

shift_right <- function(mat, n) {
  col_total <- ncol(mat)
  # 取模处理避免无效移位
  n <- n %% col_total
  if (n == 0) {
    return(mat)
  }
  # 右移n次的列顺序：最后n列放到最前面，剩余列依次拼接
  new_col_order <- c(seq(col_total - n + 1, col_total), seq(1, col_total - n))
  return(mat[, new_col_order])
}

该函数运行结果和重复调用原有shift函数的效果完全一致，矩阵越大、移动次数越多时性能优势越明显。

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