scipy中的csr_matrix

原文链接:https://muzhan.blog.csdn.net/article/details/102889734
scipy中有个csr_matrix,可以简便地表达大规模稀疏矩阵。打个比方,对于1000*1000的二维矩阵,怎么表达呢?用一个numpy array去存储吗?其实,如果这个矩阵非常稀疏,比如特征矩阵等,可以用更简便的csr matrix去表达。

from scipy.sparse import csr_matrix
x = [0,0,0,0,1,1,1,1,2,2,2,2,3,3,3,3]
y = [0,1,2,3,0,1,2,3,0,1,2,3,0,1,2,3]
value = [1,0,3,0,0,0,0,1,0,9,1,0,0,0,0,1]
mat = csr_matrix((value,(x,y)),shape=(4,4))
print(mat)
print(mat.todense())

输出为:

  (0, 0)        1
  (0, 1)        0
  (0, 2)        3
  (0, 3)        0
  (1, 0)        0
  (1, 1)        0
  (1, 2)        0
  (1, 3)        1
  (2, 0)        0
  (2, 1)        9
  (2, 2)        1
  (2, 3)        0
  (3, 0)        0
  (3, 1)        0
  (3, 2)        0
  (3, 3)        1
[[1 0 3 0]
 [0 0 0 1]
 [0 9 1 0]
 [0 0 0 1]]

上面展示的是一个4x4情况下的csr matrix的表达形式。

你可能感兴趣的:(python,csr_matrix,scipy)