题解 | 实现压缩列稀疏矩阵

import numpy as np
from scipy.sparse import csc_matrix
def compressed_col_sparse_matrix(dense_matrix):
    dense_np=np.array(dense_matrix)
    csc=csc_matrix(dense_np)
    vals=csc.data.tolist()
    row_idx=csc.indices.tolist()
    col_ptr=csc.indptr.tolist()
    return vals, row_idx, col_ptr


if __name__ == "__main__":
    dense_matrix = eval(input())
    vals, row_idx, col_ptr = compressed_col_sparse_matrix(dense_matrix)
    print(vals)
    print(row_idx)
    print(col_ptr)