python - 知道传递给函数的参数是向量还是矩阵

Question

我正在 Sage 中编写一个函数，该函数应该以不同的方式用于向量和矩阵。

我不能使用isinstance函数，因为向量或矩阵的类型取决于元素的类型：

sage: type(matrix([[1]]))
<type 'sage.matrix.matrix_integer_dense.Matrix_integer_dense'>
sage: type(matrix([[i]]))
<type 'sage.matrix.matrix_symbolic_dense.Matrix_symbolic_dense'>

区分向量和矩阵的最佳方法是什么？

Answer 1

尝试matrix.dim在 Sage 源中查找定义时意外找到了解决方案。

from sage.matrix.matrix import is_Matrix
from sage.structure.element import is_Vector

def myfunction(x):
    if is_Vector(x):
        # do something
    elif is_Matrix(x):
        # do something else
    else:
        raise TypeError("The argument must be vector or matrix")