python文件读取生成有权有向图

用字典嵌套的形式生成有权有向图,读取txt文件中的数据,给定的txt文件格式如下,中间用空格分隔开。生成的举例如下:

{1: {2: 2, 3: 6}, 3: {4: 7}, 2: {4: 3}}

 具体实现代码如下

class Graph_all_paths():
    def __init__(self):
      self.file_path = 'x.txt'  # 图文件
      self.total_eage#总边数
      self.graph = {}#用来存储图
    

    def initial_graph(self, file_path):
        f = open(file_path)
        lines = [l.split() for l in f.readlines() if l.strip()]
        # print(lines)
        self.total_eage = len(lines)  # 原始图的总线路
        for i in lines:
            if eval(i[0]) not in self.graph.keys():
                self.graph[eval(i[0])] = {eval(i[1]): eval(i[2])}
            else:
                self.graph[eval(i[0])][eval(i[1])] = eval(i[2])
        return self.graph#返回的图即为所求

我相信你应该会使用类的~

猜你喜欢

转载自blog.csdn.net/weixin_63676550/article/details/130179357
今日推荐