迭代计算scipy矩阵乘法

时间:2018-03-15 18:10:11

标签: python scipy sparse-matrix

我有一个sparse_csr矩阵c和d:

a=np.array([[1,0,1],[0,0,1]])
b=np.array([[1,1,1],[1,0,1],[0,0,0]])

c=sparse.csr_matrix(a)
d=sparse.csr_matrix(b)

我正在迭代地将c和d相乘:

for j in range(3):
    c_new = d.dot(c.T).T
    c = c_new

我发现type(c)<class 'scipy.sparse.csc.csc_matrix'>。这减慢了在每次迭代内对大矩阵进行矩阵计算。有没有办法加快保持“csr”格式的过程。

0 个答案:

没有答案