标签： python c++ scipy
我有一个稀疏的大矩阵（线性尺寸为2 * 10 ^ 6），我想为其计算轨迹。 计算它的蛮力需要16秒才能访问每个对角线元素（因此，我可以在一年内做到这一点！）。
我当时正在考虑使用scipy.io.mmwrite将其保存到磁盘，并使用应该会更快的c ++代码读取它。但是，我找不到任何可以帮助我做到这一点的软件包。
任何建议将不胜感激！
谢谢。
答案 0 :(得分：0)
解决方案非常简单：np.array（Mat.diagonal（））。sum（）
感谢@hpaulj！