Matplotlib和Networkx - 绘制自环节点

Question

Matplotlib和Networkx - 绘制自环节点

13

我有这个函数，我想画一个自环。我该怎么做？
边缘存在，但我认为在这个例子中它只是一个点(1,1)，我无法添加节点的名称。我的目标是从邻接矩阵绘制图形。是否有更好的方法来做到这一点？

import networkx as nx
import matplotlib.pyplot as plt
from matplotlib.patches import FancyArrowPatch, Circle
import numpy as np

def draw_network(G,pos,ax,sg=None):

    for n in G:
        c=Circle(pos[n],radius=0.05,alpha=0.7)
        ax.add_patch(c)
        G.node[n]['patch']=c
        x,y=pos[n]
    seen={}
    for (u,v,d) in G.edges(data=True):
        n1=G.node[u]['patch']
        n2=G.node[v]['patch']
        rad=0.1
        if (u,v) in seen:
            rad=seen.get((u,v))
            rad=(rad+np.sign(rad)*0.1)*-1
        alpha=0.5
        color='k'

        e = FancyArrowPatch(n1.center,n2.center,patchA=n1,patchB=n2,
                            arrowstyle='-|>',
                            connectionstyle='arc3,rad=%s'%rad,
                            mutation_scale=10.0,
                            lw=2,
                            alpha=alpha,
                            color=color)
        seen[(u,v)]=rad
        ax.add_patch(e)
    return e


G=nx.MultiDiGraph([(1,2),(1,1),(1,2),(2,3),(3,4),(2,4),
                (1,2),(1,2),(1,2),(2,3),(3,4),(2,4)]
                )

pos=nx.spring_layout(G)
ax=plt.gca()
draw_network(G,pos,ax)
ax.autoscale()
plt.axis('equal')
plt.axis('off')

plt.show()

- Athman3

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Bonlenfum · Accepted Answer

看起来你对matplotlib的使用相当高级，但我仍然建议使用专门的图形绘制库（就像networkx文档中所述）。随着图形变得越来越大，会出现更多问题——但这些问题在那些库中已经得到解决。

"go-to"选项是graphviz，它可以很好地处理多图的绘制。你可以从networkx图中编写dot文件，然后使用其中一个图形绘制工具（如dot、neato等）。

这里有一个例子，基于graph attributes和multigraph edge attributes：

import networkx as nx
from networkx.drawing.nx_agraph import to_agraph 

# define the graph as per your question
G=nx.MultiDiGraph([(1,2),(1,1),(1,2),(2,3),(3,4),(2,4), 
    (1,2),(1,2),(1,2),(2,3),(3,4),(2,4)])

# add graphviz layout options (see https://dev59.com/OlkS5IYBdhLWcg3wlnvg#39662097)
G.graph['edge'] = {'arrowsize': '0.6', 'splines': 'curved'}
G.graph['graph'] = {'scale': '3'}
                                                                                
# adding attributes to edges in multigraphs is more complicated but see
# https://dev59.com/jV8d5IYBdhLWcg3wkSwV#26694158                    
G[1][1][0]['color']='red'
                                                                                
A = to_agraph(G) 
A.layout('dot')                                                                 
A.draw('multi.png')

请注意，您还可以轻松地从ipython shell中调用绘图： https://dev59.com/vmUp5IYBdhLWcg3w665C#14945560