Python SciPy csgraph.breadth_first_tree用法及代碼示例

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

用法:

scipy.sparse.csgraph.breadth_first_tree(csgraph, i_start, directed=True)#

返回廣度優先搜索生成的樹

請注意，來自指定節點的廣度優先樹是唯一的。

參數 ：：

csgraph： 數組 或稀疏矩陣

表示壓縮稀疏圖的 N x N 矩陣。輸入的 csgraph 將被轉換為 csr 格式進行計算。

i_start： int

起始節點的索引。

directed： 布爾型，可選

如果為 True(默認值)，則對有向圖進行操作：僅沿著路徑 csgraph[i, j] 從點 i 移動到點 j。如果為 False，則在無向圖上找到最短路徑：算法可以沿著 csgraph[i, j] 或 csgraph[j, i] 從點 i 前進到 j。

返回 ：：

cstree： 企業社會責任矩陣

從 csgraph 繪製的廣度優先樹的 N x N 有向 compressed-sparse 表示，從指定節點開始。

注意：

如果可能存在多個有效解決方案，則輸出可能會因 SciPy 和 Python 版本而異。

例子：

以下示例顯示了在簡單的 four-component 圖上計算深度優先樹(從節點 0 開始)：

input graph          breadth first tree from (0)

     (0)                         (0)
    /   \                       /   \
   3     8                     3     8
  /       \                   /       \
(3)---5---(1)               (3)       (1)
  \       /                           /
   6     2                           2
    \   /                           /
     (2)                         (2)

在壓縮稀疏表示中，解決方案如下所示：

>>> from scipy.sparse import csr_matrix
>>> from scipy.sparse.csgraph import breadth_first_tree
>>> X = csr_matrix([[0, 8, 0, 3],
...                 [0, 0, 2, 5],
...                 [0, 0, 0, 6],
...                 [0, 0, 0, 0]])
>>> Tcsr = breadth_first_tree(X, 0, directed=False)
>>> Tcsr.toarray().astype(int)
array([[0, 8, 0, 3],
       [0, 0, 2, 0],
       [0, 0, 0, 0],
       [0, 0, 0, 0]])

請注意，生成的圖是跨越該圖的有向無環圖。來自給定節點的廣度優先樹是唯一的。