Python：使用一组邻接列表构建图

Question

您好，这是我的第一篇文章，我的英语不是很流利，如果我做错了发帖协议，请原谅我......

所以，我现在通过邻接矩阵实现了一个图：

class GFG: 
def __init__(self,graph): 
    self.graph = graph

我是这样开始的：

my_shape = (N,M)
bpGraph = np.zeros(my_shape)

并像这样添加边：

vertex_a = 2
vertex_b = 5
bpGraph [vertex_a, vertex_b] = 1
g = GFG(bpGraph)

并像这样检查边缘：

if bpGraph [vertice_a, vertice_b]:
do something...

但是这种结构如果我有很多顶点的话会占用很多空间，所以我想切换到set map

class GFG: 
def __init__(self,N,M):
    mysetdict = {
        "dummy_a": {"dummy_b","dummy_c"},
    }
    self.graph = mysetdict 

def add_edges(self,N,M):
    N_set = {}
    if str(N) in self.graph:
        N_set = self.graph[str(N)]
    N_set.add(str(M))

    self.graph[str(N)] = N_set

def check_edges(self,N,M):
     return str(M) in self.graph[str(N)]

当我尝试添加一个顶点时，我得到了这个错误：

g = GFG (3,4)
g.add_edges (0, 0)
AttributeError: 'dict' object has no attribute 'add'
---> 16                 N_set.add(str(M))

我尝试了很多方法来构建这个集合，并在添加一个 vertex_a 以使用 vertex_b 做一条边之前检查是否存在一个条目，但是我所有的尝试都会导致属性错误。

有什么建议吗？

Answer 1

解决了，只做N_set = set()

def add_edges(self,N,M):
    N_set = set()
    if str(N) in self.m_graph:
        N_set = self.m_graph[str(N)]
    N_set.add(str(M))

    self.m_graph[str(N)] = N_set