scipy.sparse.csgraph.csgraph_to_masked

scipy.sparse.csgraph.csgraph_to_masked(csgraph)

Convert a sparse graph representation to a masked array representation

New in version 0.11.0.

Parameters:

csgraphcsr_matrix, csc_matrix, or lil_matrix

Sparse representation of a graph.

Returns:

graphMaskedArray

The masked dense representation of the sparse graph.

Examples

Try it in your browser!

>>> from scipy.sparse import csr_matrix
>>> from scipy.sparse.csgraph import csgraph_to_masked

>>> graph = csr_matrix( [
... [0, 1, 2, 0],
... [0, 0, 0, 1],
... [0, 0, 0, 3],
... [0, 0, 0, 0]
... ])
>>> graph
<4x4 sparse matrix of type '<class 'numpy.int64'>'
    with 4 stored elements in Compressed Sparse Row format>

>>> csgraph_to_masked(graph)
masked_array(
  data=[[ --, 1.0, 2.0,  --],
        [ --,  --,  --, 1.0],
        [ --,  --,  --, 3.0],
        [ --,  --,  --,  --]],
  mask=[[ True, False, False,  True],
        [ True,  True,  True, False],
        [ True,  True,  True, False],
        [ True,  True,  True,  True]],
  fill_value=1e+20)

Go BackOpen In Tab