python稀疏矩阵scipy.sparse中的csr_matrix和csc_matrix的详细解析

一、

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])
csr_matrix((data, indices, indptr), shape=(3, 3)).toarray()

输出:
array([[1, 0, 2],
       [0, 0, 3],
       [4, 5, 6]])
csr中的r表示row,即为行优先。

indptr中的元素两个两个的看,前两个为0和2,表示第一行有2-0=2个元素,再在indices中选前两个元素,得到在列上的位置为0和2,填充data的元素为1和2。接在是indptr中的2和3,所以第二行有3-2=1个元素,要填充的列位置是indices 中的2,元素从data中找为3. 在接下来是indptr中的3和6,表示第三行有6-3=3个元素,要填充的位置是indices 中的0、1、2,元素从data中找为4,5,6.

二、

 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])
csc_matrix((data, indices, indptr), shape=(3, 3)).toarray()

输出:
array([[1, 0, 4],
       [0, 0, 5],
       [2, 3, 6]])

这里的csc中的c表示的是column,表示列优先。原来和上面的类似。
 

发布了141 篇原创文章 · 获赞 54 · 访问量 15万+

猜你喜欢

转载自blog.csdn.net/iamjingong/article/details/97292613