Python SciPy sparse.block_diag用法及代碼示例

本文簡要介紹 python 語言中 scipy.sparse.block_diag 的用法。

用法:

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

從提供的矩陣構建塊對角稀疏矩陣或數組。

參數 ：：

mats： 矩陣或數組的序列

輸入矩陣或數組。

format： str，可選

結果的稀疏格式(例如“csr”)。如果未給出，結果將以“coo” 格式返回。

dtype： dtype 說明符，可選

輸出的數據類型。如果未給出，則數據類型由塊的數據類型確定。

返回 ：：

res： 稀疏矩陣或數組

如果至少一個輸入是稀疏數組，則輸出也是稀疏數組。否則輸出是稀疏矩陣。

注意：

例子：

>>> 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]])