,[c@sdZddlZddlmZddlmZmZmZm Z m Z m Z ddddd d d d d dddg Z e e dZe eedZe dddde e e e e e e d Ze dddddde e e e ee de ddZe ddddde e d Ze d!ddddde e ed" Zd#Zd$Zd%Zd&Zd'Zd(Ze e e d)Zd*ZdS(+s ********** Matplotlib ********** Draw networks with matplotlib. See Also -------- matplotlib: http://matplotlib.org/ pygraphviz: http://pygraphviz.github.io/ iN(tis_string_like(t shell_layouttcircular_layouttkamada_kawai_layouttspectral_layoutt spring_layoutt random_layouttdrawt draw_networkxtdraw_networkx_nodestdraw_networkx_edgestdraw_networkx_labelstdraw_networkx_edge_labelst draw_circulartdraw_kamada_kawait draw_randomt draw_spectralt draw_springt draw_shellcKsyddlj}Wn5tk r2tdntk rJdGHnX|dkrf|j}n |j}|jd|dkr|jdkr|j d }q|j }nd|krd |k|d>> G = nx.dodecahedral_graph() >>> nx.draw(G) >>> nx.draw(G, pos=nx.spring_layout(G)) # use spring layout See Also -------- draw_networkx() draw_networkx_nodes() draw_networkx_edges() draw_networkx_labels() draw_networkx_edge_labels() Notes ----- This function has the same name as pylab.draw and pyplot.draw so beware when using >>> from networkx import * since you might overwrite the pylab.draw function. With pyplot use >>> import matplotlib.pyplot as plt >>> import networkx as nx >>> G = nx.dodecahedral_graph() >>> nx.draw(G) # networkx draw() >>> plt.draw() # pyplot draw() Also see the NetworkX drawing examples at https://networkx.github.io/documentation/latest/auto_examples/index.html iNsMatplotlib required for draw()s!Matplotlib unable to open displaytwiit with_labelstlabelstpostax(iiii(tmatplotlib.pyplottpyplott ImportErrort RuntimeErrortNonetgcft get_figuret set_facecolort_axstacktadd_axestgcaRt set_axis_offtdraw_if_interactive(tGRRtkwdstplttcf((s8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pyR,s0;        cKsyddlj}Wn5tk r2tdntk rJdGHnX|dkrltjj|}nt|||}t ||d||}|rt |||n|j dS(s)Draw the graph G using Matplotlib. Draw the graph with Matplotlib with options for node positions, labeling, titles, and many other drawing features. See draw() for simple drawing without labels or axes. Parameters ---------- G : graph A networkx graph pos : dictionary, optional A dictionary with nodes as keys and positions as values. If not specified a spring layout positioning will be computed. See :py:mod:`networkx.drawing.layout` for functions that compute node positions. arrows : bool, optional (default=True) For directed graphs, if True draw arrowheads. Note: Arrows will be the same color as edges. arrowstyle : str, optional (default='-|>') For directed graphs, choose the style of the arrowsheads. See :py:class: `matplotlib.patches.ArrowStyle` for more options. arrowsize : int, optional (default=10) For directed graphs, choose the size of the arrow head head's length and width. See :py:class: `matplotlib.patches.FancyArrowPatch` for attribute `mutation_scale` for more info. with_labels : bool, optional (default=True) Set to True to draw labels on the nodes. ax : Matplotlib Axes object, optional Draw the graph in the specified Matplotlib axes. nodelist : list, optional (default G.nodes()) Draw only specified nodes edgelist : list, optional (default=G.edges()) Draw only specified edges node_size : scalar or array, optional (default=300) Size of nodes. If an array is specified it must be the same length as nodelist. node_color : color string, or array of floats, (default='r') Node color. Can be a single color format string, or a sequence of colors with the same length as nodelist. If numeric values are specified they will be mapped to colors using the cmap and vmin,vmax parameters. See matplotlib.scatter for more details. node_shape : string, optional (default='o') The shape of the node. Specification is as matplotlib.scatter marker, one of 'so^>v>> G = nx.dodecahedral_graph() >>> nx.draw(G) >>> nx.draw(G, pos=nx.spring_layout(G)) # use spring layout >>> import matplotlib.pyplot as plt >>> limits = plt.axis('off') # turn of axis Also see the NetworkX drawing examples at https://networkx.github.io/documentation/latest/auto_examples/index.html See Also -------- draw() draw_networkx_nodes() draw_networkx_edges() draw_networkx_labels() draw_networkx_edge_labels() iNsMatplotlib required for draw()s!Matplotlib unable to open displaytarrows( RRRRRtnxtdrawingRR R R R$(R%RR)RR&R'tnode_collectiontedge_collection((s8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pyRs   i,trtog?cKsddl}yddlj}ddl}Wn5tk rJtdntk rbdGHnX| dkr~|j} n|dkrt|}n| st |dkrdSy*|j g|D]}||^q}WnEt k r}t j d|n tk r't j dnXt||jr^t|||||| }d}n| j|dddf|dddfd |d |d |d |d |d| d|d| d| d|  }|jd|S(sU Draw the nodes of the graph G. This draws only the nodes of the graph G. Parameters ---------- G : graph A networkx graph pos : dictionary A dictionary with nodes as keys and positions as values. Positions should be sequences of length 2. ax : Matplotlib Axes object, optional Draw the graph in the specified Matplotlib axes. nodelist : list, optional Draw only specified nodes (default G.nodes()) node_size : scalar or array Size of nodes (default=300). If an array is specified it must be the same length as nodelist. node_color : color string, or array of floats Node color. Can be a single color format string (default='r'), or a sequence of colors with the same length as nodelist. If numeric values are specified they will be mapped to colors using the cmap and vmin,vmax parameters. See matplotlib.scatter for more details. node_shape : string The shape of the node. Specification is as matplotlib.scatter marker, one of 'so^>v>> G = nx.dodecahedral_graph() >>> nodes = nx.draw_networkx_nodes(G, pos=nx.spring_layout(G)) Also see the NetworkX drawing examples at https://networkx.github.io/documentation/latest/auto_examples/index.html See Also -------- draw() draw_networkx() draw_networkx_edges() draw_networkx_labels() draw_networkx_edge_labels() iNsMatplotlib required for draw()s!Matplotlib unable to open displayisNode %s has no position.sBad value in node positions.itstctmarkertcmaptvmintvmaxtalphat linewidthst edgecolorstlabeli(t collectionsRRtnumpyRRRR"tlisttlentasarraytKeyErrorR*t NetworkXErrort ValueErrort isinstancetIterablet apply_alphatscattert set_zorder(R%Rtnodelistt node_sizet node_colort node_shapeR6R3R4R5RR7R8R9R&R:R'tnptvtxyteR,((s8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pyR sJZ     *  /  tktsolids-|>i c@ snyvddl}ddlj}ddlj}ddlm}m}m}ddl m }ddl m }ddl Wn5tk rtdntk rdGHnX| dkr|j} n|dkrt|j}n| st|dkrdS|dkr(t|j}njg|D]"}||d||d f^q5}|j|s{|f}n|}t| r|j|rt|t|krjg|D]}t|^qr tg|D]}|j||^q}qjg|D]}t| ^qrjg|D]'}|j|oet|dk^qArt|}qd}qtd nHt|st|d kr|j||f}nd } t| |j s| r||d|d|ddd|d| j}!|!jd |!j || j!|!|j"|rg|!j#|n|dkr| dk rt$| |st%n|!j&j||!j'| | dk s| dk r|!j(| | q|!j)n|!Sd}"|jr>| r>fd}#g}"|}$|}%|%dkr| dk r_t$| |skt%n |j*} | dkrt+|} n| dkrt,|} n|d| d| }&nxt-|D]q\}'\}(})|(\}*}+|)\},}-d}.d}/d}0d}1|j|rR||'\}2}3|j.|3}4||4}5|#|5|}1n|#||}1|%dkr| |&||'}.n)t|%d kr|%|'}.n |%d}.t|d kr||'}/n |d}/||*|+f|,|-fd|d|0d|1d|$d|.d|/dd }6|"j/|6| j0|6qWnj1j2|dddddf}7j3j2|dddddf}8j1j2|ddddd f}9j3j2|ddddd f}:|8|7};|:|9}<d|;d|<}=}>|7|=|9|>f|8|=|:|>ff}?| j4|?| j5|"S( s Draw the edges of the graph G. This draws only the edges of the graph G. Parameters ---------- G : graph A networkx graph pos : dictionary A dictionary with nodes as keys and positions as values. Positions should be sequences of length 2. edgelist : collection of edge tuples Draw only specified edges(default=G.edges()) width : float, or array of floats Line width of edges (default=1.0) edge_color : color string, or array of floats Edge color. Can be a single color format string (default='r'), or a sequence of colors with the same length as edgelist. If numeric values are specified they will be mapped to colors using the edge_cmap and edge_vmin,edge_vmax parameters. style : string Edge line style (default='solid') (solid|dashed|dotted,dashdot) alpha : float The edge transparency (default=1.0) edge_ cmap : Matplotlib colormap Colormap for mapping intensities of edges (default=None) edge_vmin,edge_vmax : floats Minimum and maximum for edge colormap scaling (default=None) ax : Matplotlib Axes object, optional Draw the graph in the specified Matplotlib axes. arrows : bool, optional (default=True) For directed graphs, if True draw arrowheads. Note: Arrows will be the same color as edges. arrowstyle : str, optional (default='-|>') For directed graphs, choose the style of the arrow heads. See :py:class: `matplotlib.patches.ArrowStyle` for more options. arrowsize : int, optional (default=10) For directed graphs, choose the size of the arrow head head's length and width. See :py:class: `matplotlib.patches.FancyArrowPatch` for attribute `mutation_scale` for more info. label : [None| string] Label for legend Returns ------- matplotlib.collection.LineCollection `LineCollection` of the edges list of matplotlib.patches.FancyArrowPatch `FancyArrowPatch` instances of the directed edges Depending whether the drawing includes arrows or not. Notes ----- For directed graphs, arrows are drawn at the head end. Arrows can be turned off with keyword arrows=False. Be sure to include `node_size` as a keyword argument; arrows are drawn considering the size of nodes. Examples -------- >>> G = nx.dodecahedral_graph() >>> edges = nx.draw_networkx_edges(G, pos=nx.spring_layout(G)) >>> G = nx.DiGraph() >>> G.add_edges_from([(1, 2), (1, 3), (2, 3)]) >>> arcs = nx.draw_networkx_edges(G, pos=nx.spring_layout(G)) >>> alphas = [0.3, 0.4, 0.5] >>> for i, arc in enumerate(arcs): # change alpha values of arcs ... arc.set_alpha(alphas[i]) Also see the NetworkX drawing examples at https://networkx.github.io/documentation/latest/auto_examples/index.html See Also -------- draw() draw_networkx() draw_networkx_nodes() draw_networkx_labels() draw_networkx_edge_labels() iN(tcolorConvertertColormapt Normalize(tLineCollection(tFancyArrowPatchsMatplotlib required for draw()s!Matplotlib unable to open displayiiiis.edge_color must contain color names or numberss8edge_color must be a color or list of one color per edgetcolorsR7t antialiasedst linestylet transOffsetcs6|dkr!jd|dSj|dSdS(Nss^>vtiterableRtalltruettupletto_rgbaRAt is_directedt transDataRFt set_labeltadd_collectiont is_numliket set_alphaRBtAssertionErrort set_arraytset_cmaptset_climt autoscaletget_cmaptmintmaxt enumeratetindextappendt add_patchtamintraveltamaxtupdate_datalimtautoscale_view(@R%Rtedgelisttwidtht edge_colortstyleR6R]t arrowsizet edge_cmapt edge_vmint edge_vmaxRR)R9RHRGRJR&RdR'tcbRQRRRSRTRURNtedge_postlwR1t edge_colorstmsgR-tarrow_collectionR\R`t arrow_colorst color_normaltitsrctdsttx1ty1tx2ty2t arrow_colort line_widtht shrink_sourcet shrink_targettsrc_nodetdst_nodet index_nodeR[tarrowtminxtmaxxtminytmaxyRthtpadxtpadytcorners((RKs8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pyR sr      8  (() 4                          1111  (  i s sans-seriftnormalc Ksty"ddlj} ddlj} Wn5tk rAtdntk rYdGHnX| dkru| j} n|dkrtd|j D}n| j dd} | j dd}i}x|j D]\}}||\}}t |st |}n| j|||d |d |d |d |d |d| d|d| jd|dt }|||>> G = nx.dodecahedral_graph() >>> labels = nx.draw_networkx_labels(G, pos=nx.spring_layout(G)) Also see the NetworkX drawing examples at https://networkx.github.io/documentation/latest/auto_examples/index.html See Also -------- draw() draw_networkx() draw_networkx_nodes() draw_networkx_edges() draw_networkx_edge_labels() iNsMatplotlib required for draw()s!Matplotlib unable to open displaycss|]}||fVqdS(N((t.0tn((s8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pys sthorizontalalignmenttcentertverticalalignmenttsizeRatfamilytweightR6t transformtbboxtclip_on(RRReRfRRRR"tdictRktgettitemsRtstrttextRqtTrue(R%RRt font_sizet font_colort font_familyt font_weightR6RRR&R'RRRt text_itemsRR9txtytt((s8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pyR sBA        g?c Ksyddlj} ddl}Wn5tk r>tdntk rVdGHnX| dkrr| j} n|dkrd|jdtD}n|}i}x|j D]\\}}}||\}}||\}}|||d||||d|}}| r|j ||||d|j d }|d kr]|d 8}n|d krv|d 7}n|j ||f}| j j|j |f|jd"d}nd}| dkrtdddd#dd$} nt|st|}n| jdd}| jdd}| j|||d|d|d|d|d|d|d|d|d| j d| d d d!t }||||f>> G = nx.dodecahedral_graph() >>> edge_labels = nx.draw_networkx_edge_labels(G, pos=nx.spring_layout(G)) Also see the NetworkX drawing examples at https://networkx.github.io/documentation/latest/auto_examples/index.html See Also -------- draw() draw_networkx() draw_networkx_nodes() draw_networkx_edges() draw_networkx_labels() iNsMatplotlib required for draw()s!Matplotlib unable to open displaycSs(i|]\}}}|||fqS(((RtuRLtd((s8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pys s tdatag?g@ihiZiiiiigtboxstyletroundtectfcRRRRRaRRR6trotationRRRcR(ii(g?g?g?(g?g?g?(RRR;RRRR"RjRRtarctan2tpitarrayRqttransform_anglestreshapeRRRRR( R%Rt edge_labelst label_posRRRRR6RRtrotateR&R'RKRRtn1tn2R9RRRRRRtangleRMt trans_angleRRR((s8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pyR /sjL    )           cKst|t||dS(s8Draw the graph G with a circular layout. Parameters ---------- G : graph A networkx graph kwargs : optional keywords See networkx.draw_networkx() for a description of optional keywords, with the exception of the pos parameter which is not used by this function. N(RR(R%tkwargs((s8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pyR s cKst|t||dS(sKDraw the graph G with a Kamada-Kawai force-directed layout. Parameters ---------- G : graph A networkx graph kwargs : optional keywords See networkx.draw_networkx() for a description of optional keywords, with the exception of the pos parameter which is not used by this function. N(RR(R%R((s8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pyRs cKst|t||dS(s6Draw the graph G with a random layout. Parameters ---------- G : graph A networkx graph kwargs : optional keywords See networkx.draw_networkx() for a description of optional keywords, with the exception of the pos parameter which is not used by this function. N(RR(R%R((s8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pyRs cKst|t||dS(s8Draw the graph G with a spectral layout. Parameters ---------- G : graph A networkx graph kwargs : optional keywords See networkx.draw_networkx() for a description of optional keywords, with the exception of the pos parameter which is not used by this function. N(RR(R%R((s8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pyRs cKst|t||dS(s6Draw the graph G with a spring layout. Parameters ---------- G : graph A networkx graph kwargs : optional keywords See networkx.draw_networkx() for a description of optional keywords, with the exception of the pos parameter which is not used by this function. N(RR(R%R((s8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pyRs cKsH|jdd}|dk r(|d=nt|t|d||dS(s6Draw networkx graph with shell layout. Parameters ---------- G : graph A networkx graph kwargs : optional keywords See networkx.draw_networkx() for a description of optional keywords, with the exception of the pos parameter which is not used by this function. tnlistN(RRRR(R%RR((s8lib/python2.7/site-packages/networkx/drawing/nx_pylab.pyRs   cCs+ddl}ddlm}m}y/ddl} ddlm} ddlj} Wnt k rpt dnXt |t |krt |d|j r| j d|} | j||| j|} n^y| j| j|g} Wn<tk r0| jg|D]}| j|^q} nXyt |t | ksa| jt |kr| jt |df| d | d ddf<| d | d dd f<| d| d dd fs . Z }  d      R