【问题标题】:Calculating the trace of a large sparse matrix计算大型稀疏矩阵的迹
【发布时间】:2019-01-26 14:10:45
【问题描述】:

我有一个稀疏的大矩阵(线性维度为 2*10^6),我想计算它的迹线。 蛮力计算需要 16 秒才能访问每个对角线元素(因此我可以在一年内完成!)。

我正在考虑使用 scipy.io.mmwrite 将其保存到磁盘并使用应该更快的 c++ 代码读取它。但是我找不到任何可以帮助我做到这一点的包。

任何建议将不胜感激!

谢谢。

【问题讨论】:

  • 它是什么格式的?
  • 这是一个 csc_matrix
  • 16 秒在 python 中?
  • 每个元素,是的。
  • M.diagonal().sum() 工作吗?

标签: python c++ scipy


【解决方案1】:

解决方案很简单:np.array(Mat.diagonal()).sum()

谢谢@hpaulj!

【讨论】:

    猜你喜欢
    • 2015-09-14
    • 2021-05-09
    • 1970-01-01
    • 1970-01-01
    • 2014-10-11
    • 2018-06-04
    • 2021-09-22
    • 2015-03-01
    • 2013-06-09
    相关资源
    最近更新 更多