Python 3: TypeError: 'int' object is not iterable on a list of string elements

Python 3: TypeError: 'int' object is not iterable on a list of string elements

我正在尝试实现一个由邻接表表示的图形数据结构。我正在使用以顶点作为键的字典,每个键可以有一个或多个值。值由一个列表组成,该列表包含键顶点连接到的另一个顶点,以及在顶点(无向图)之间遍历的成本。由于一个顶点可以连接到其他几个顶点,因此某个键的值可以由嵌套列表组成。

我知道我的实现不是很面向对象,我的代码可以在很多方面改进,但我想暂时保持这样,因为当我从非常糟糕的事情开始并改进时,我会学得更好之后。

我在使用函数 is_connected(self, node1, node2) 时遇到问题。我试图通过首先将每个键的值复制到两个单独的列表来查看两个顶点是否连接(在邻接列表中)。然后我对每个列表执行列表推导,以便展平任何嵌套列表。这个想法是检查每个顶点是否在彼此的值列表中,从而得出它们是否连接的结论。

但是,在尝试进行列表理解时会出现问题。

class UndirectedGraph():
    def __init__(self):
        self.vertices = my_dict = {
                                    'A' : ['B', 10],
                                    'B' : [['A', 10], ['C', 5]],
                                    'C' : [['B', 5], ['D', 15]],
                                    'D' : [['C', 15], ['E', 25]],
                                    'E' : ['D', 25]
                                                    }
    def __str__(self):
        return str(self.vertices)

    def add(self, new_node):
        if new_node in self.vertices:
            return False
        else:
            self.vertices[new_node] = []
            return True

    def connect(self, node1, node2, cost):
        if node1 in self.vertices and node2 in self.vertices:
            if self.is_connected(node1, node2):
                for x, y in [(x, y) for x in self.vertices.get(node1) for y in self.vertices.get(node2)]:
                    if node2 in x and node1 in y:
                        x[1] = cost
                        y[1] = cost
            else:
                if node1 == node2:
                    node1_values = [node2, cost]
                    self.vertices[node1].append(node1_values)
                else:
                    node1_values = [node2, cost]
                    self.vertices[node1].append(node1_values)
                    node2_values = [node1, cost]
                    self.vertices[node2].append(node2_values)

    def is_connected(self, node1, node2):
        node1_values = self.vertices[node1]
        node2_values = self.vertices[node2]
        n1 = [str(item) for sublist in node1_values for item in sublist]
        n2 = [str(item) for sublist in node2_values for item in sublist]

        if node2 in n1 and node1 in n2:
            print(node1, "and", node2, "are connected")
            return True
        else:
            print(node1, "and", node2, "are not connected")
            return False

我创建了 n 个 class 实例并调用方法 is_connected(self, node1, node2)

g = UndirectedGraph()
g.is_connected('A', 'B')

我收到以下错误:

Traceback (most recent call last):
  File ".\UndirectedGraph.py", line 83, in <module>
    g.is_connected('A', 'B')
  File ".\UndirectedGraph.py", line 53, in is_connected
    n1 = [str(item) for sublist in node1_values for item in sublist]
  File ".\UndirectedGraph.py", line 53, in <listcomp>
    n1 = [str(item) for sublist in node1_values for item in sublist]
TypeError: 'int' object is not iterable

我怀疑这与值列表中的某些元素是整数这一事实有关,这就是为什么在开始时尝试将它们转换为字符串对象如果列表理解,但我猜他们是仍被解释为整数。

我在想也许我必须在列表理解之前进行整数到字符串的转换,但是除了 .join( ) 方法 ,我不想使用它,因为我希望能够将两个不同列表的元素相互比较。如果我使用 .join() 方法,这意味着我必须将所有值添加到一个字符串中,然后将字符串拆分为一个列表?

如何解决这个问题?

提前致谢! 或者有没有

比较这两个节点(来自您的 self.vertices dict):

'A' : ['B', 10],
'B' : [['A', 10], ['C', 5]],

我假设这应该表示节点 'A' 有一条到节点 'B' 的边,成本为 10 并且节点 'B' 有相同的边加上另一条边到节点 'C' 成本为 5。本质上,您的 vertices 字典应该从节点名称(例如 'A')映射到 "list of edges with their cost"。但是,对于节点 'A',您忘记包含外部列表:dict-value 不是嵌套列表。修复很简单:

        # in you __init__ function
        self.vertices = my_dict = {
                                    'A' : [['B', 10]],
                                    'B' : [['A', 10], ['C', 5]],
                                    'C' : [['B', 5], ['D', 15]],
                                    'D' : [['C', 15], ['E', 25]],
                                    'E' : [['D', 25]]
                                                    }

通过此修复,我得到:

>>> g = UndirectedGraph()
>>> g.is_connected('A', 'B')
('A', 'and', 'B', 'are connected')