在R中将矩阵转换为累积行均值矩阵

Question

我有一组预测矩阵，每行代表一个个体的预测结果，每列代表一个特定模型的预测结果。我想把它转换成第一列是第一个模型的预测，第二列是第一和第二模型预测结果的平均值等等。

因此，转换后的矩阵将包含原始矩阵中观察结果的累加平均值。

我感觉可以使用cumsum函数结合apply函数来实现这个目标，但不确定如何得到优雅的结果（适用于大型矩阵）。

谢谢！

- Rocinante

Answer 1

试试这个：

# Initialize a testing matrix
(m <- matrix(1:12, 3, 4))

     [,1] [,2] [,3] [,4]
[1,]    1    4    7   10
[2,]    2    5    8   11
[3,]    3    6    9   12

# Calculate cumulative average by column for each row
t(apply(m, 1, cumsum) / seq(ncol(m)))

     [,1] [,2] [,3] [,4]
[1,]    1  2.5    4  5.5
[2,]    2  3.5    5  6.5
[3,]    3  4.5    6  7.5

这基本上是对行累加求和，然后除以一个表示列索引的循环数组。

编辑：如果您正在使用数据框架进行类似操作，则可以使用data.table和reshape2包来实现：

library(data.table)
dt <- data.table(m)
# Add row number to melt by
dt[, row := seq(nrow(dt))]

library(reshape2)
dt.molten <- data.table(melt(dt, "row"))
# Row-level format
dt.molten[, cumsum(value) / as.numeric(variable), "row"]

    row  V1                                                                                          
 1:   1 1.0                                                                                          
 2:   1 2.5                                                                                          
 3:   1 4.0                                                                                          
 4:   1 5.5                                                                                          
 5:   2 2.0                                                                                          
 6:   2 3.5                                                                                          
 7:   2 5.0                                                                                          
 8:   2 6.5                                                                                          
 9:   3 3.0                                                                                          
10:   3 4.5                                                                                          
11:   3 6.0
12:   3 7.5

- Max Ghenis

Answer 2

使用建议的 cumsum 和 apply。

mat <- matrix(1:24,ncol=6)
mat
#     [,1] [,2] [,3] [,4] [,5] [,6]
#[1,]    1    5    9   13   17   21
#[2,]    2    6   10   14   18   22
#[3,]    3    7   11   15   19   23
#[4,]    4    8   12   16   20   24

t(apply(mat,1,cumsum)/(seq_len(ncol(mat))))
#     [,1] [,2] [,3] [,4] [,5] [,6]
#[1,]    1    3    5    7    9   11
#[2,]    2    4    6    8   10   12
#[3,]    3    5    7    9   11   13
#[4,]    4    6    8   10   12   14

- thelatemail