Python稀疏矩阵运算

import numpy as np
import scipy
import time
import scipy.sparse as sparse

t = [1]+[0]*4999
a = scipy.matrix(np.array(t*5000, dtype=float).reshape(5000, 5000))
start = time.time()
b = np.dot(a, a)
print(time.time() - start)

start = time.time()
aa = sparse.coo_matrix(a)
b = np.dot(aa, aa).todense()
print(time.time() - start)

 

 

参考：

[1] https://zhuanlan.zhihu.com/p/52497385