scipy.sparse.
block_diag#
- scipy.sparse.block_diag(mats, format=None, dtype=None)[源码]#
从提供的矩阵构建块对角稀疏矩阵或数组。
- 参数:
- mats矩阵或数组序列
输入矩阵或数组。
- format字符串,可选
结果的稀疏格式(例如,“csr”)。如果未指定,结果将以“coo”格式返回。
- dtype数据类型指定符,可选
输出的数据类型。如果未指定,数据类型将根据 blocks 的类型确定。
- 返回:
- res稀疏矩阵或数组
如果至少一个输入是稀疏数组,则输出为稀疏数组。否则,输出为稀疏矩阵。
另请参阅
备注
在 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]])