Python SciPy sparse.triu用法及代码示例

本文简要介绍 python 语言中 scipy.sparse.triu 的用法。

用法:

scipy.sparse.triu(A, k=0, format=None)#

以稀疏格式返回矩阵的上三角部分

返回矩阵 A 的 k-th 对角线之上或之上的元素。：

• k = 0 对应于主对角线

• k > 0 在主对角线上方

• k < 0 在主对角线下方

参数 ：：

A： 密集或稀疏矩阵

需要其上三角部分的矩阵。

k： 整数 可选的

上三角形最底部的对角线。

format： string

结果的稀疏格式，例如格式=”csr”等

返回 ：：

L： 稀疏矩阵

稀疏格式的 A 的上三角部分。

例子：

>>> from scipy.sparse import csr_matrix, triu
>>> A = csr_matrix([[1, 2, 0, 0, 3], [4, 5, 0, 6, 7], [0, 0, 8, 9, 0]],
...                dtype='int32')
>>> A.toarray()
array([[1, 2, 0, 0, 3],
       [4, 5, 0, 6, 7],
       [0, 0, 8, 9, 0]])
>>> triu(A).toarray()
array([[1, 2, 0, 0, 3],
       [0, 5, 0, 6, 7],
       [0, 0, 8, 9, 0]])
>>> triu(A).nnz
8
>>> triu(A, k=1).toarray()
array([[0, 2, 0, 0, 3],
       [0, 0, 0, 6, 7],
       [0, 0, 0, 9, 0]])
>>> triu(A, k=-1).toarray()
array([[1, 2, 0, 0, 3],
       [4, 5, 0, 6, 7],
       [0, 0, 8, 9, 0]])
>>> triu(A, format='csc')
<3x5 sparse matrix of type '<class 'numpy.int32'>'
        with 8 stored elements in Compressed Sparse Column format>