Python中如何创建无节点顺序偏见且兼容扩展的Edge对象？

问题核心是无向边的双向连通性没被正确处理，同时要预留有向边、多重边的扩展空间，我们可以从Edge类的封装和Network的邻接逻辑入手：

1. 给Edge类添加通用的节点关联方法 #

给Edge增加判断节点归属、获取邻接节点的方法，同时预留有向边的扩展标记，不管初始化时节点顺序如何，都能正确识别关联关系：

class Edge:
    def __init__(self, node1, node2, directed=False):
        self.p1 = node1
        self.p2 = node2
        # 新增directed标记，默认无向，后续扩展有向边直接修改这个参数即可
        self.directed = directed

    def contains_node(self, node):
        # 判断当前节点是否属于这条边
        return node == self.p1 or node == self.p2

    def get_adjacent(self, node):
        # 根据给定节点返回邻接节点，无向边双向支持，有向边后续可扩展逻辑
        if node == self.p1:
            return self.p2
        elif node == self.p2 and not self.directed:
            return self.p1
        # 有向边场景下，若节点是p2则返回None，按需调整
        return None

2. 修正Network的邻接节点查询逻辑 #

修改adjacent_nodes方法，通过Edge的封装方法获取邻接节点，彻底摆脱对节点顺序的依赖：

class Network:
    def __init__(self, nodes = [], edges = []):
        self.nodes = nodes
        self.edges = edges

    def adjacent_nodes(self, node):
        adjacent = []
        for edge in self.edges:
            neighbor = edge.get_adjacent(node)
            if neighbor is not None:
                adjacent.append(neighbor)
        return adjacent

    # 原maximal_subnetwork方法无需修改，因为依赖的adjacent_nodes已经修正
    def maximal_subnetwork(self, node):
        nodes = {node}
        traced = set()
        while nodes:
            node = nodes.pop()
            traced.add(node)
            for i in self.adjacent_nodes(node):
                    if i not in traced:
                        nodes.add(i)
        traced = list(traced)
        return Network(nodes = traced , edges = self.return_edges(*traced))

    # 补充实现return_edges方法（原代码未实现，这里给出通用版本）
    def return_edges(self, *nodes):
        node_set = set(nodes)
        return [edge for edge in self.edges if edge.p1 in node_set and edge.p2 in node_set]

3. 为什么这样处理？ #

• 通用性拉满：通过directed标记，后续要做有向边时，只需要调整get_adjacent的逻辑（比如有向边只允许从p1到p2的邻接）；多重边场景下，因为遍历所有边，天然支持同节点对的多条边，完全不需要重构现有结构。
• 彻底消除顺序依赖：不管Edge初始化时node1和node2的顺序是啥，contains_node和get_adjacent都能正确识别节点关联，解决孤立节点误判的问题。
• 封装更合理：节点关联的逻辑都封装在Edge类内部，Network只需要调用方法即可，符合面向对象的设计原则。

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