Python SciPy csgraph.csgraph_from_masked用法及代码示例

本文简要介绍 python 语言中 scipy.sparse.csgraph.csgraph_from_masked 的用法。

用法:

scipy.sparse.csgraph.csgraph_from_masked(graph)#

从掩码数组构造 CSR-format 图。

参数 ：：

graph： MaskedArray

输入图。形状应该是(n_nodes，n_nodes)。

返回 ：：

csgraph： csr_matrix

图的压缩稀疏表示，

例子：

>>> import numpy as np
>>> from scipy.sparse.csgraph import csgraph_from_masked

>>> graph_masked = np.ma.masked_array(data =[
... [0, 1, 2, 0],
... [0, 0, 0, 1],
... [0, 0, 0, 3],
... [0, 0, 0, 0]
... ],
... mask=[[ True, False, False,  True],
...       [ True,  True,  True, False],
...       [ True,  True,  True, False],
...       [ True,  True,  True,  True]],
... fill_value = 0)

>>> csgraph_from_masked(graph_masked)
<4x4 sparse matrix of type '<class 'numpy.float64'>'
    with 4 stored elements in Compressed Sparse Row format>