diff options
Diffstat (limited to 'networkx/drawing/nx_pylab.py')
| -rw-r--r-- | networkx/drawing/nx_pylab.py | 22 |
1 files changed, 20 insertions, 2 deletions
diff --git a/networkx/drawing/nx_pylab.py b/networkx/drawing/nx_pylab.py index e01b2054..836ff85e 100644 --- a/networkx/drawing/nx_pylab.py +++ b/networkx/drawing/nx_pylab.py @@ -540,9 +540,15 @@ def draw_networkx_edges( floats from 0-1. If numeric values are specified they will be mapped to colors using the edge_cmap and edge_vmin,edge_vmax parameters. - style : string (default=solid line) + style : string or array of strings (default='solid') Edge line style e.g.: '-', '--', '-.', ':' or words like 'solid' or 'dashed'. + Can be a single style or a sequence of styles with the same + length as the edge list. + If less styles than edges are given the styles will cycle. + If more styles than edges are given the styles will be used sequentially + and not be exhausted. + Also, `(offset, onoffseq)` tuples can be used as style instead of a strings. (See `matplotlib.patches.FancyArrowPatch`: `linestyle`) alpha : float or None (default=None) @@ -822,6 +828,18 @@ def draw_networkx_edges( else: line_width = width + if ( + np.iterable(style) + and not isinstance(style, str) + and not isinstance(style, tuple) + ): + if len(style) == len(edge_pos): + linestyle = style[i] + else: # Cycle through styles + linestyle = style[i % len(style)] + else: + linestyle = style + arrow = mpl.patches.FancyArrowPatch( (x1, y1), (x2, y2), @@ -832,7 +850,7 @@ def draw_networkx_edges( color=arrow_color, linewidth=line_width, connectionstyle=_connectionstyle, - linestyle=style, + linestyle=linestyle, zorder=1, ) # arrows go behind nodes |