scipy 稀疏矩阵的类型提示

Question

如何输入提示 scipy 稀疏矩阵，例如CSR, CSC, LIL等？以下是我一直在做的事情，但感觉不对：

def foo(mat: scipy.sparse.csr.csr_matrix):
    # Do whatever

如果我们的函数可以接受多种类型的 scipy 稀疏矩阵（即其中任何一个），我们该怎么办？

Answer 1

scipy.sparse.base现已弃用。您应该使用scipy.sparse.spmatrix它，它为所有稀疏矩阵提供基类。

def foo(mat: scipy.sparse.spmatrix):
    # Whatever you want
    pass