如何在Python中高效地添加稀疏矩阵

Question

我想知道如何在Python中高效地添加稀疏矩阵。

我有一个程序，它将一个大任务分解成子任务，并将它们分布到几个CPU上。每个子任务产生一个结果（一个scipy稀疏矩阵，格式为：lil_matrix）。

稀疏矩阵的维数是：10000x500000，这是相当大的，所以我真的需要最有效的方法来将所有得到的稀疏矩阵求和为一个稀疏矩阵，使用一些C编译的方法或其他方法。

Answer 1

你试过最简单的计时方法吗？

matrix_result = matrix_a + matrix_b

文档警告说，对于LIL矩阵，这可能很慢，这表明以下操作可能更快：

matrix_result = (matrix_a.tocsr() + matrix_b.tocsr()).tolil()