python - 需要 NetworkX 帮助

Question

目前我面临以下问题:

我有一个脚本，可以搜索包含文档的特定目录。每个文档的文件名中都分配有一个编号。每个文档中的数字也代表另一个文档(文件名)。如何创建一个网络来显示哪些文档会导致什么结果？

如有任何帮助，我们将不胜感激，谢谢

Answer 1

这是有向图的教科书示例。您应该阅读NetworkX tutorial更好地了解如何与他们合作；基本上，您需要添加所有节点(点)，在本例中为文件编号，然后在它们之间添加边。

import os
import networkx as nx

g = nx.DiGraph( )
for filename in os.listdir( <dir> ):
    # do something to filename to get the number
    g.add_node( <number> )

for filename in os.listdir( <dir> ):
    # do something to filename to get the source
    with open( filename ) as theFile:
        # do something to theFile to get the targets
        for target in <targets>:
            g.add_edge( <source>, <target> )

import matplotlib.pyplot as plt
nx.draw( g )