python“ zip”,并循环创建对象

时间:2019-11-17 14:38:08

标签: python sorting object zip cycle

我刚刚学习了python中的zip命令,我想将其用于拓扑排序。但是,我的代码仍然有一些问题。

我有此python脚本用于拓扑排序:

from collections import defaultdict


class Graph:
    def __init__(self, directed=False):
        self.graph = defaultdict(list)
        self.directed = directed

    def addEdge(self, frm, to):
        self.graph[frm].append(to)

        if self.directed is False:
            self.graph[to].append(frm)
        else:
            self.graph[to] = self.graph[to]

    def topoSortvisit(self, s, visited, sortlist):
        visited[s] = True

        for i in self.graph[s]:
            if not visited[i]:
                self.topoSortvisit(i, visited, sortlist)

        sortlist.insert(0, s)

    def topoSort(self):
        visited = {i: False for i in self.graph}

        sortlist = []

        for v in self.graph:
            if not visited[v]:
                self.topoSortvisit(v, visited, sortlist)

我还有我的csv文件:

name;id;linked;distance;type
A;1;4;60807.66;N
B;2;3;50949.67;N
C;3;4;39067.50;N
D;4;5;48169.93;N
E;5;0;0;N

正如我刚从另一篇文章中学到的,我将执行以下操作:

dfm = pd.read_csv(fname, sep=';',index_col=None, header = 0)
zipl  = zip(dfm['id'], dfm['linked'])

根据dfm ['id'],dfm ['linked']我想对它们进行排序。因此,我将Graph对象创建为:

links = Graph(directed=True)

并对它们进行排序:

[links.addEdge(row) for row in zipl]

但是,出现以下错误:

TypeError: addEdge() missing 1 required positional argument: 'to' 

如果我没有问题的话,

links.addEdge(4, 5)

这是有关数据类型的问题吗?

0 个答案:

没有答案
相关问题