有向图创建python代码
时间: 2024-03-11 08:43:02 浏览: 19
在Python中,可以使用邻接表或邻接矩阵来表示有向图,并通过相应的数据结构来创建有向图。下面是使用邻接表表示有向图的Python代码示例:
```python
class DirectedGraph:
def __init__(self):
self.graph = {}
def add_edge(self, start, end):
if start in self.graph:
self.graph[start].append(end)
else:
self.graph[start] = [end]
def get_neighbors(self, node):
if node in self.graph:
return self.graph[node]
else:
return []
# 创建有向图对象
graph = DirectedGraph()
# 添加边
graph.add_edge(1, 2)
graph.add_edge(1, 3)
graph.add_edge(2, 3)
graph.add_edge(3, 4)
# 获取节点的邻居
neighbors = graph.get_neighbors(1)
print(neighbors) # 输出:[2, 3]
```
上述代码中,`DirectedGraph`类表示有向图,使用字典`graph`来存储节点和其对应的邻居节点。`add_edge`方法用于添加边,`get_neighbors`方法用于获取节点的邻居。