如何从两个用户之间的连接表中检索已连接的用户集群？在Python中

# 连接表示例数据
connection_table = [
    ['UserA', 'UserB'],
    ['UserB', 'UserC'],
    ['UserD', 'UserE'],
    ['UserF', 'UserG'],
    ['UserG', 'UserH'],
    ['UserI', 'UserJ'],
    ['UserK', 'UserL'],
    ['UserM', 'UserN'],
    ['UserO', 'UserP'],
    ['UserP', 'UserQ']
]

def retrieve_connected_clusters(connection_table):
    clusters = []  # 存储已连接用户集群的列表

    for connection in connection_table:
        user_a, user_b = connection[0], connection[1]

        cluster_a = None
        cluster_b = None

        # 检查用户A是否已经存在于某个集群中
        for cluster in clusters:
            if user_a in cluster:
                cluster_a = cluster
                break

        # 检查用户B是否已经存在于某个集群中
        for cluster in clusters:
            if user_b in cluster:
                cluster_b = cluster
                break

        if cluster_a is None and cluster_b is None:
            # 创建一个新的集群，并将用户A和用户B添加进去
            new_cluster = {user_a, user_b}
            clusters.append(new_cluster)
        elif cluster_a is None:
            # 将用户A添加到用户B所在的集群中
            cluster_b.add(user_a)
        elif cluster_b is None:
            # 将用户B添加到用户A所在的集群中
            cluster_a.add(user_b)
        elif cluster_a != cluster_b:
            # 将用户B所在的集群合并到用户A所在的集群中
            cluster_a.update(cluster_b)
            clusters.remove(cluster_b)

    return clusters

# 调用函数并打印结果
result = retrieve_connected_clusters(connection_table)
print(result)