extract sparse rows from sparse matrix in r

2019-07-26 07:34发布

问题:

I have a large sparse matrix to analyze in R. For instance:

i <- c(1,3:8); j <- c(2,9,6:10); x <- 7 * (1:7)
(A <- sparseMatrix(i, j, x = x))
[1,] . 7 . . .  .  .  .  .  .
[2,] . . . . .  .  .  .  .  .
[3,] . . . . .  .  .  . 14  .
[4,] . . . . . 21  .  .  .  .
[5,] . . . . .  . 28  .  .  .
[6,] . . . . .  .  . 35  .  .
[7,] . . . . .  .  .  . 42  .
[8,] . . . . .  .  .  .  . 49

I want to extract the i-th row from this matrix, as a sparse vector. If I write

(x=A[1,])

I obtain

 [1] 0 7 0 0 0 0 0 0 0 0

but what I would like is

 [1] . 7 . . . . . . . .

What I expect is that the new vector does not materialize the zeros. How can I do this?

Thanks

回答1:

You can use drop = FALSE:

A[1, , drop = FALSE]
# 1 x 10 sparse Matrix of class "dgCMatrix"
#                        
# [1,] . 7 . . . . . . . .