将 1 和 0 的矩阵转换为行和计数矩阵

Question

我想将 0 和 1 的矩阵转换为相应的矩阵，该矩阵给出非零项的累积行总和。输入和输出示例如下：

set.seed(404)
input  <- matrix(rbinom(10 * 5, 1, 0.5), ncol = 5, nrow = 5)
output <- data.frame(a = c(1, 1, 1, 1, 0),
                     b = c(0, 0, 0, 0, 0),
                     c = c(2, 2, 0, 2, 1),
                     d = c(3, 0, 0, 3, 2),
                     e = c(0, 3, 0, 0, 0))

input
#     [,1] [,2] [,3] [,4] [,5]
#[1,]    1    0    1    1    0
#[2,]    1    0    1    0    1
#[3,]    1    0    0    0    0
#[4,]    1    0    1    1    0
#[5,]    0    0    1    1    0
output
#  a b c d e
#1 1 0 2 3 0
#2 1 0 2 0 3
#3 1 0 0 0 0
#4 1 0 2 3 0
#5 0 0 1 2 0

Answer 1

我们可以用apply和MARGIN=1得到'input'每一行的cumsum，转置（t）和[=相乘24=] 以便 1 值被 cumsum 输出替换，'0' 保持不变。

input*t(apply(input, 1, cumsum))
#   [,1] [,2] [,3] [,4] [,5]
#[1,]    1    0    2    3    0
#[2,]    1    0    2    0    3
#[3,]    1    0    0    0    0
#[4,]    1    0    2    3    0
#[5,]    0    0    1    2    0

或者我们可以使用 library(matrixStats) 中的 rowCumsums 得到每一行的 cumsum 并像以前一样相乘。

library(matrixStats)
input*rowCumsums(input)
#     [,1] [,2] [,3] [,4] [,5]
#[1,]    1    0    2    3    0
#[2,]    1    0    2    0    3
#[3,]    1    0    0    0    0
#[4,]    1    0    2    3    0
#[5,]    0    0    1    2    0

将 1 和 0 的矩阵转换为行和计数矩阵

Convert a matrix of 1s and 0s to a row-sum count matrix

r

matrix

rowsum