【发布时间】:2019-01-26 14:10:45
【问题描述】:
我有一个稀疏的大矩阵(线性维度为 2*10^6),我想计算它的迹线。 蛮力计算需要 16 秒才能访问每个对角线元素(因此我可以在一年内完成!)。
我正在考虑使用 scipy.io.mmwrite 将其保存到磁盘并使用应该更快的 c++ 代码读取它。但是我找不到任何可以帮助我做到这一点的包。
任何建议将不胜感激!
谢谢。
【问题讨论】:
-
它是什么格式的?
-
这是一个 csc_matrix
-
16 秒在 python 中?
-
每个元素,是的。
-
M.diagonal().sum()工作吗?