如何在 python 中查看使用图形生成的网络？

Question

我正在尝试生成网络并希望以图形方式查看它。

我正在使用一个列表来存储边缘信息。

我尝试了以下方法：

  from collections import defaultdict
class Graph(object):
    def __init__(self,connections):
        self.graph = defaultdict(set)
        self.addConnections(connections)

    def addConnections(self, connections):
        for node1, node2 in connections:
            self.addEdge(node1, node2)

    def addEdge(self, node1, node2):
        self.graph[node1].add(node2)

    def generateEdges(self): 
        edges = [] 
        for node in self.graph: 
            for neighbour in self.graph[node]: 
                edges.append((node, neighbour))
        self.edges = edges
        return edges

我正在尝试按如下方式查看图表：

import networkx as nx
import matplotlib.pyplot as plt
from graph import Graph
#create edges for first graph
graph1Edges = [('A','B'), ('A', 'D'), ('B', 'C'), ('C','D')]
graph1 = Graph(graph1Edges)
nx.draw(graph1.generateEdges())
plt.show()

但是，它不起作用，请建议我一种方法，因为我不想使用 networkx 包。

Answer 1

您尝试自己实现一个 Graph 并使用 networkx 显示它，而不是使用 networkx 的图形实现。这当然是不可能的，因为 nx.draw 需要一个 networkx 图。

只需删除您的 Graph 实现并运行：

import networkx as nx
import matplotlib.pyplot as plt

edges = [('A', 'B'), ('A', 'D'), ('B', 'C'), ('C', 'D')]
G = nx.Graph()
G.add_edges_from(edges)
nx.draw_networkx(G)
plt.show()

你会得到：