我刚刚学习了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)
这是有关数据类型的问题吗?