检查上三角矩阵或下三角矩阵

Question

在使用numpy或scipy时，有没有办法检查矩阵是否为下三角矩阵或上三角矩阵？我知道如何编写一个函数来检查这一点；但我想知道这些模块是否有自己的函数。我在文档中搜索但没有找到任何信息。

- Juan

Answer 1

我会做

np.allclose(mat, np.tril(mat)) # check if lower triangular
np.allclose(mat, np.triu(mat)) # check if upper triangular
np.allclose(mat, np.diag(np.diag(mat))) # check if diagonal

• http://docs.scipy.org/doc/numpy/reference/generated/numpy.tril.html
• http://docs.scipy.org/doc/numpy/reference/generated/numpy.triu.html
• http://docs.scipy.org/doc/numpy/reference/generated/numpy.allclose.html
• http://docs.scipy.org/doc/numpy/reference/generated/numpy.diag.html

- YXD

---

太棒了！还要检查对角线吗？ - Juan

有没有避免复制的方法来实现这个？ - eickenberg

我想不到避免使用numpy复制的方法。如果这是代码中的瓶颈，我会尝试使用cython或numba编写测试。 - YXD