码迷,mamicode.com
首页 > 其他好文 > 详细

scipy构建稀疏矩阵

时间:2018-11-10 00:09:35      阅读:264      评论:0      收藏:0      [点我收藏+]

标签:array   nump   class   mat   csr   numpy   div   sci   rom   

from scipy.sparse import csr_matrix
import numpy as np
indptr  = np.array([0, 2, 3, 6])
indices = np.array([0, 2, 2, 0, 1, 2]) 
data    = np.array([1, 2, 3, 4, 5, 6]) #表示要构建稀疏矩阵的数据


#按照行来压缩,
#方法:第i行(本例中i=0,1,2),
#非零数据列的索引为indices[indptr[i]:indptr[i+1]] 
#非零数据列的数据为   data[indptr[i]:indptr[i+1]]
csr_matrix((data, indices, indptr),shape=(3, 3)).toarray()
array([[1, 0, 2],
       [0, 0, 3],
       [4, 5, 6]])
from scipy.sparse import csc_matrix
#按照列来压缩,
#方法:第i列(本例中i=0,1,2),
#非零数据行的索引为indices[indptr[i]:indptr[i+1]] 
#非零数据行的数据为   data[indptr[i]:indptr[i+1]]

csc_matrix((data, indices, indptr), shape=(3, 3)).toarray()
#和上面的例子正好是转置
array([[1, 0, 4],
       [0, 0, 5],
       [2, 3, 6]])

scipy构建稀疏矩阵

标签:array   nump   class   mat   csr   numpy   div   sci   rom   

原文地址:https://www.cnblogs.com/wzdLY/p/9937699.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!