You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何在R语言中不使用for循环实现矩阵按固定列组求和

实现矩阵按3列分组求和(无for循环)

当然可以不用for循环搞定这个需求!R的向量化操作和矩阵重塑特性完全能实现,而且效率比循环高得多。下面给你两种实用方法,都不需要写for循环:

方法1:分组拆分+行求和(直观易读)

先构造一个示例矩阵方便演示:

set.seed(123)
# 生成3行12列的示例矩阵(列数为3的倍数)
mat <- matrix(sample(1:20, 3*12, replace = TRUE), nrow = 3)
print(mat)

接下来我们把矩阵的列按每3个一组拆分,对每组内的列按行求和,最后合并结果:

# 按每3列分组,计算每组的行和后合并成新矩阵
grouped_sum <- do.call(cbind, 
                       lapply(split(seq(ncol(mat)), ceiling(seq(ncol(mat))/3)), 
                              function(col_indices) rowSums(mat[, col_indices])))
print(grouped_sum)

方法2:矩阵重塑+列求和(高效向量化)

这个方法完全依赖R的向量化矩阵操作,处理大矩阵时速度会更快:

# 转置原矩阵→拆分为3行的块→对每列求和→转置回原行数
grouped_sum_fast <- t(colSums(matrix(t(mat), nrow = 3)))
print(grouped_sum_fast)

原理拆解:

  1. t(mat)把原矩阵转置,让原来的列变为行;
  2. matrix(t(mat), nrow=3)将转置后的矩阵重新排列为3行,这样原矩阵的每3列就对应新矩阵的1列;
  3. colSums对新矩阵的每列求和,得到每组3列的行和;
  4. 最后用t()转置回来,就得到了符合需求的矩阵。

补充:处理列数非3倍数的情况

如果你的矩阵列数不是3的整数倍,比如有10列,可以调整分组逻辑,比如用floor(seq(ncol(mat))/3)把最后不足3列的部分单独作为一组求和,根据你的实际需求灵活调整即可。

内容的提问来源于stack exchange,提问作者Pete

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.04.29 14:57:35