r - 根据单独向量中索引处的值合并列

Question

我有一个这样的矩阵:

> y
     [,1] [,2] [,3] [,4]
[1,]   17   14    5    8
[2,]    7    2   17    2
[3,]   10   18    6    6
[4,]    6   11    4   11
[5,]    5    9    4    9

和一个向量

> group
[1] 1 2 2 3

组向量表示第 1 列在第 1 组中，第 2 列和第 3 列在第 2 组中，第 3 列在第 3 组中。我想合并 y 中的列(通过加法)，以便分配给特定组的所有列都是相加即得到:

     [,1] [,2] [,3]
[1,]   17   19    8
[2,]    7   19    2
[3,]   10   24    6
[4,]    6   15   11
[5,]    5   13    9

我一直在搞弄子集、合并和减少，但我真的一无所获。

Answer 1

我们可以做的

sapply(split(seq_along(group), group), function(x) rowSums(y[, x, drop = FALSE]))
#     1  2  3
#[1,] 17 19  8
#[2,]  7 19  2
#[3,] 10 24  6
#[4,]  6 15 11
#[5,]  5 13  9

或者另一个选项是 rowsum

t(rowsum(t(y), group))
#      1  2  3
#[1,] 17 19  8
#[2,]  7 19  2
#[3,] 10 24  6
#[4,]  6 15 11
#[5,]  5 13  9