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>