2016-12-28 54 views
-4

我正在使用python实现Dijkstra算法,我想知道如何从列表中移除一个对象的实例? 下面是我的代码中出现错误的一部分:“list.remove(x):x not in list”。如何从python列表中删除对象的实例?

class Vertex: 
    def __init__(self, id, name): 
     self.id = id 
     self.name = name 
     self.minDistance = float("inf") 
     self.previousVertex = None 
     self.edges = [] 

    def computePath(self, sourceId): 
     for i in self.vertexes: 
      if i.id == sourceId: 
       startVertex = i 
       startVertex.minDistance=0 
       break 
      else: 
       continue 
     unvisited = [] 
     for vertex in self.vertexes: 
      unvisited.append(vertex) 
     while len(unvisited)!=0: 
      self.visited.append(startVertex) 
      unvisited.remove(startVertex) 
+0

您一直尝试在循环中反复移除'startVertex'。 – user2357112

+0

另外'pop()'是一个很好的函数,可以从列表中删除事物,因为它已经到位了。 – MooingRawr

+0

@MooingRawr也是这样''.remove' –

回答

1

list.remove()将从列表,该项目是等于任何你想要删除删除的第一个项目。

换句话说,该算法是这样做:那么

def remove(self, item_to_remove): 
    for index, item in enumerate(self): 
     if item == item_to_remove: 
      del self[index] 
      return 
    raise ValueError('list.remove(x): x not in list') 

诀窍是让==平等试验合格。这是通过__eq__ method完成的。 默认的实现是只返回True当两个项目是一个和同一个对象,所以item is item_to_remove是真的。

您可以为您Vertex类中创建一个自定义的方法:

def __eq__(self, other): 
    if not isinstance(other, Vertex): 
     return NotImplemented 
    return self.id == other.id # refine as needed 

你需要决定当两个Vertex实例是相等的。