Python Data Structure 简明教程

Python - Graphs

图是对一组对象的图形化表示,其中某些对象对通过链接连接。互连的对象由称为顶点的点表示,连接顶点的链接称为边。与图相关的各种术语和功能在本教程中有详细说明。

在本章中,我们将看到如何使用 Python 程序创建图,并向其中添加各种数据元素。以下是我们在图上进行的基本操作。

  1. Display graph vertices

  2. Display graph edges

  3. Add a vertex

  4. Add an edge

  5. Creating a graph

可以使用 Python 字典数据类型轻松呈现图。我们将顶点表示为字典的键,并将称为边的顶点之间的连接表示为字典中的值。

请看下面的图表 −

graph basics

在上图中,

V = {a, b, c, d, e}
E = {ab, ac, bd, cd, de}

Example

我们可以将此图表示为以下 Python 程序 −

# Create the dictionary with graph elements
graph = {
   "a" : ["b","c"],
   "b" : ["a", "d"],
   "c" : ["a", "d"],
   "d" : ["e"],
   "e" : ["d"]
}
# Print the graph
print(graph)

Output

执行上述代码后,将生成以下结果 −

{'c': ['a', 'd'], 'a': ['b', 'c'], 'e': ['d'], 'd': ['e'], 'b': ['a', 'd']}

Display graph vertices

要显示图顶点,我们只需找到图字典的键。我们使用 keys() 方法。

class graph:
   def __init__(self,gdict=None):
      if gdict is None:
         gdict = []
      self.gdict = gdict
# Get the keys of the dictionary
   def getVertices(self):
      return list(self.gdict.keys())
# Create the dictionary with graph elements
graph_elements = {
   "a" : ["b","c"],
   "b" : ["a", "d"],
   "c" : ["a", "d"],
   "d" : ["e"],
   "e" : ["d"]
}
g = graph(graph_elements)
print(g.getVertices())

Output

执行上述代码后,将生成以下结果 −

['d', 'b', 'e', 'c', 'a']

Display graph edges

找到图边比找到顶点要困难一些,因为我们必须找到每对之间有边的顶点的各个对。因此,我们创建一个空的边列表,然后遍历与每个顶点关联的边值。我们将形成一个包含从顶点找到的不同组边的列表。

class graph:
   def __init__(self,gdict=None):
      if gdict is None:
         gdict = {}
      self.gdict = gdict

   def edges(self):
      return self.findedges()
# Find the distinct list of edges
   def findedges(self):
      edgename = []
      for vrtx in self.gdict:
         for nxtvrtx in self.gdict[vrtx]:
            if {nxtvrtx, vrtx} not in edgename:
               edgename.append({vrtx, nxtvrtx})
      return edgename
# Create the dictionary with graph elements
graph_elements = {
   "a" : ["b","c"],
   "b" : ["a", "d"],
   "c" : ["a", "d"],
   "d" : ["e"],
   "e" : ["d"]
}
g = graph(graph_elements)
print(g.edges())

Output

执行上述代码后,将生成以下结果 −

[{'b', 'a'}, {'b', 'd'}, {'e', 'd'}, {'a', 'c'}, {'c', 'd'}]

Adding a vertex

添加顶点非常直接,我们在图字典中添加另一个额外的键即可。

Example

class graph:
   def __init__(self,gdict=None):
      if gdict is None:
         gdict = {}
      self.gdict = gdict
   def getVertices(self):
      return list(self.gdict.keys())
# Add the vertex as a key
   def addVertex(self, vrtx):
      if vrtx not in self.gdict:
         self.gdict[vrtx] = []
# Create the dictionary with graph elements
graph_elements = {
   "a" : ["b","c"],
   "b" : ["a", "d"],
   "c" : ["a", "d"],
   "d" : ["e"],
   "e" : ["d"]
}
g = graph(graph_elements)
g.addVertex("f")
print(g.getVertices())

Output

执行上述代码后,将生成以下结果 −

['a', 'b', 'c', 'd', 'e','f']

Adding an edge

向现有图添加边涉及将新顶点视为元组,并验证该边是否已经存在。如果不存在,则添加该边。

class graph:
   def __init__(self,gdict=None):
      if gdict is None:
         gdict = {}
      self.gdict = gdict
   def edges(self):
      return self.findedges()
# Add the new edge
   def AddEdge(self, edge):
      edge = set(edge)
      (vrtx1, vrtx2) = tuple(edge)
      if vrtx1 in self.gdict:
         self.gdict[vrtx1].append(vrtx2)
      else:
         self.gdict[vrtx1] = [vrtx2]
# List the edge names
   def findedges(self):
      edgename = []
      for vrtx in self.gdict:
         for nxtvrtx in self.gdict[vrtx]:
            if {nxtvrtx, vrtx} not in edgename:
               edgename.append({vrtx, nxtvrtx})
        return edgename
# Create the dictionary with graph elements
graph_elements = {
   "a" : ["b","c"],
   "b" : ["a", "d"],
   "c" : ["a", "d"],
   "d" : ["e"],
   "e" : ["d"]
}
g = graph(graph_elements)
g.AddEdge({'a','e'})
g.AddEdge({'a','c'})
print(g.edges())

Output

执行上述代码后,将生成以下结果 −

[{'e', 'd'}, {'b', 'a'}, {'b', 'd'}, {'a', 'c'}, {'a', 'e'}, {'c', 'd'}]