Python图类实现：遍历Node对象无法调用类方法的问题求助

看起来你遇到的问题是遍历Graph里的Node对象时，只能看到对象类型<class '__main__.Node'>，没法看到自定义输出或者调用Node的方法对吧？这通常是因为你没给Node类实现__str__或__repr__方法，或者在遍历的时候没有正确调用对象的方法。

我给你写一个完整的示例，你可以参考：

首先是Node类的实现，这里必须加上__str__和__repr__来定义对象的显示格式：

class Node:
    def __init__(self, value):
        self.value = value  # 给Node加个标识值，方便区分节点
        self.neighbors = []  # 存储当前节点的邻居节点
    
    def add_neighbor(self, node):
        """添加邻居节点，避免重复添加"""
        if node not in self.neighbors:
            self.neighbors.append(node)
    
    def __str__(self):
        """定义直接打印Node对象时的友好输出格式"""
        return f"Node(value={self.value})"
    
    def __repr__(self):
        """定义Node对象在列表等容器中显示时的格式，直接复用__str__的结果即可"""
        return self.__str__()

然后是Graph类，负责管理所有Node对象：

class Graph:
    def __init__(self):
        self.nodes = []  # 存储图中的所有节点
    
    def add_node(self, value):
        """创建并添加节点到图中"""
        new_node = Node(value)
        self.nodes.append(new_node)
        return new_node
    
    def get_all_nodes(self):
        """返回所有节点，方便外部遍历"""
        return self.nodes

接下来是测试代码，看看实际效果：

# 构建测试图
my_graph = Graph()
node_a = my_graph.add_node("A")
node_b = my_graph.add_node("B")
node_c = my_graph.add_node("C")

# 添加邻居关系
node_a.add_neighbor(node_b)
node_a.add_neighbor(node_c)

# 遍历所有节点
print("图中的所有节点:", my_graph.get_all_nodes())
# 输出: 图中的所有节点: [Node(value=A), Node(value=B), Node(value=C)]

# 遍历node_a的邻居
print("节点A的邻居:", node_a.neighbors)
# 输出: 节点A的邻居: [Node(value=B), Node(value=C)]

# 直接调用Node的方法或打印单个节点
print("单个节点输出:", node_b)
# 输出: 单个节点输出: Node(value=B)

关键点说明： #

• __str__是给print()这类直接打印对象的场景用的，定义了对象的友好字符串表示
• __repr__是给列表、字典这类容器用的，当你打印包含Node对象的容器时，Python会自动调用每个元素的__repr__方法
• 如果之前你误写了print(type(node))，那肯定只会看到<class '__main__.Node'>，要直接输出对象本身（如print(node)）或者调用对象的方法（如node.add_neighbor(xxx)）才能正常访问Node的功能

这样修改后，你遍历的时候就能看到Node的友好输出，也能正常调用Node的所有方法啦。

内容的提问来源于stack exchange，提问作者user9474716