네트워크 그래프를 그리기 위해선 우선 matplotlib를 설치해줍니다.
import matplotlib.pyplot as plt
import networkx as nx
import matplotlib.pyplot as plt
graph = nx.Graph()
graph.add_nodes_from((1,2,3,4,5))
graph.add_edges_from([(1,2), (1,3),(1,4),(3,5)])
nx.draw(graph)
plt.show()
networkX를 이용해 그래프를 만들고 draw()함수와 matplotlib의 show()함수를 이용해 그래프를 그릴 수 있습니다.
draw() 함수 이외도 draw_random(), draw_circular(), draw_spectral() 등의 함수가 있습니다.
nx.draw_circular(graph) | nx.draw_spectral(graph) |
그래프를 이미지로 저장할 수도 있습니다.
plt.savefig("networkGraph.png")
Graphviz, PyGraphviz, pydot 등이 시스템이 설치되어 있다면, 아래처럼 그래프를 그릴 수도 있습니다.
nx.draw_Graphviz(graph)
nx.write_dot(graph, 'file.dot')
반응형