将稀疏矩阵转储到文件中

时间:2011-05-22 11:11:01

标签: python csv scipy sparse-matrix file-handling

我有一个scipy.sparse.csr矩阵,想将其转储到CSV文件中。有没有办法保留矩阵的稀疏性并将其写入CSV?

1 个答案:

答案 0 :(得分:8)

SciPy包含通过scipy.io module读取/写入MatrixMarket格式的稀疏矩阵的函数,包括mmwrite:http://docs.scipy.org/doc/scipy/reference/generated/scipy.io.mmwrite.html

MatrixMarket不是CSV,而是关闭。它由一个单行标题组成,其中包含#rows,#num,#of nonzeros,后跟每个非零的一行。这些行中的每一行都是行索引,列索引,值。您可以编写一个简单的脚本,将空格转换为逗号,并且您有一个CSV。