scipy.sparse.

block_diag#

scipy.sparse.block_diag(mats, format=None, dtype=None)[source]#

從提供的矩陣建立分塊對角稀疏矩陣或陣列。

參數:
mats矩陣或陣列的序列

輸入矩陣或陣列。

format字串,選用

結果的稀疏格式 (例如,“csr”)。如果未給定,則結果以 “coo” 格式返回。

dtypedtype 指定符,選用

輸出的資料類型。如果未給定,dtype 會從 blocks 的資料類型決定。

回傳值:
res稀疏矩陣或陣列

如果至少一個輸入是稀疏陣列,則輸出為稀疏陣列。否則輸出為稀疏矩陣。

另請參閱

block_array
diags_array

註解

在版本 0.11.0 中新增。

範例

>>> from scipy.sparse import coo_array, block_diag
>>> A = coo_array([[1, 2], [3, 4]])
>>> B = coo_array([[5], [6]])
>>> C = coo_array([[7]])
>>> block_diag((A, B, C)).toarray()
array([[1, 2, 0, 0],
       [3, 4, 0, 0],
       [0, 0, 5, 0],
       [0, 0, 6, 0],
       [0, 0, 0, 7]])