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

Ale*_*ros 7 python numpy add scipy sparse-matrix

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

我有一个程序将一个大任务分解为子任务并将它们分配到多个CPU.每个子任务产生一个结果(scipy稀疏矩阵格式为:)lil_matrix.

稀疏矩阵维度是:100000x500000,这是非常巨大的,所以我真的需要最有效的方法将所有生成的稀疏矩阵求和成一个稀疏矩阵,使用一些C编译方法或其他东西.

Hug*_*ell 10

你尝试过最简单的方法吗?

matrix_result = matrix_a + matrix_b
Run Code Online (Sandbox Code Playgroud)

文档警告说,这可能对LIL矩阵来说很慢,这表明以下内容可能更快:

matrix_result = (matrix_a.tocsr() + matrix_b.tocsr()).tolil()
Run Code Online (Sandbox Code Playgroud)