{ } Raw JSON

bundles / scipy 1.18.0 / scipy / sparse / _construct / block_diag

function

scipy.sparse._construct:block_diag

source: /scipy/sparse/_construct.py:1410

Signature

def block_diag(mats, format = None, dtype = None)

Summary

Build a block diagonal sparse matrix or array from provided matrices.

Extended Summary

Parameters

mats : sequence of matrices or arrays

Input matrices or arrays.

format : str, optional

The sparse format of the result (e.g., "csr"). If not given, the result is returned in "coo" format.

dtype : dtype specifier, optional

The data-type of the output. If not given, the dtype is determined from that of blocks.

Returns

res : sparse matrix or array

If at least one input is a sparse array, the output is a sparse array. Otherwise the output is a sparse matrix.

Notes

Examples

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

See also

block_array
diags_array

Aliases

  • scipy.sparse.block_diag

Referenced by