количество ненулевых элементов по строке в R

Question

Я хочу нумеровать ненулевые элементы в matrix по строке. Вот небольшой набор данных и желаемый результат. Я бы предпочел решение в базе R.

my.data <- matrix(c(10,  0,  0,  0,  0,
                     0,  3,  9,  0,  1,
                     2, 12,  0,  0,  0,
                     5,  5,  5,  0,  5,
                     0,  0,  0,  0,  0), nrow = 5, byrow = TRUE)

desired.result <- matrix(c( 1,  0,  0,  0,  0,
                            0,  1,  2,  0,  3,
                            1,  2,  0,  0,  0,
                            1,  2,  3,  0,  4,
                            0,  0,  0,  0,  0), nrow = 5, byrow = TRUE)

Answer 1

Другая пара вариантов:

# create new matrix with multiplication
t(apply(my.data != 0, 1, cumsum)) * (my.data != 0)


# alternative:
# replace elements in original matrix
my.data[my.data != 0] = t(apply(my.data != 0, 1, cumsum))[my.data != 0]
my.data
#      [,1] [,2] [,3] [,4] [,5]
# [1,]    1    0    0    0    0
# [2,]    0    1    2    0    3
# [3,]    1    2    0    0    0
# [4,]    1    2    3    0    4
# [5,]    0    0    0    0    0

Answer 2

Вот относительно наивный базовый метод R:

t(apply(my.data, 1, function(x) {
    x[x != 0] <- seq_len(sum(x != 0))
    x
    }))

     [,1] [,2] [,3] [,4] [,5]
[1,]    1    0    0    0    0
[2,]    0    1    2    0    3
[3,]    1    2    0    0    0
[4,]    1    2    3    0    4
[5,]    0    0    0    0    0