python - 检查 scipy 稀疏矩阵是 CSC 还是 CSR

Question

我想知道检查 scipy 稀疏矩阵(CSC 或 CSR)的最佳方法。我现在正在使用。

rows, cols = X.shape()
indptr = X.indptr()
if len(indptr) == cols + 1:
    print "csc"
else:
    print "csr"

谢谢。

Answer 1

看起来您可以使用 .getformat() 方法:

>>> m0 = scipy.sparse.csc_matrix([1])
>>> m0.getformat()
'csc'
>>> m1 = scipy.sparse.csr_matrix([1])
>>> m1.getformat()
'csr'