..
Define matrix & check values
from scipy.sparse import csr_matrix
import numpy as np
row = np.array([0, 0, 1, 2, 2, 2])
col = np.array([0, 1, 2, 0, 1, 2])
data = np.array([1, 1, 1, 1, 1, 1])
mtx2 = csr_matrix((data, (row, col)), shape=(3, 3))
print(mtx2) #matrix print out
print(mtx2.toarray()) #print out by array
>
(0, 0) 1 (0, 1) 1 (1, 2) 1 (2, 0) 1 (2, 1) 1 (2, 2) 1
>
[[1 1 0] [0 0 1] [1 1 1]]..
...
get back the row, col and data value from matrix
c = mtx2.tocoo()
print(c.row)
print(c.col)
print(c.data)
>
[0 0 1 2 2 2] [0 1 2 0 1 2] [1 1 1 1 1 1]...
No comments:
Post a Comment