
    Gd                     2   d Z ddlZddlmZ ddlZg dZg dZd Z	 e	e Z
d e
                    d          D             Z ej        e
d	
          Z ej        d            ej        e
eed            ej        d            ej                     dS )z=
===================
Multipartite Layout
===================
    N)   r            r   r   r   )goldvioletr   r   r   	limegreenr	   
darkorangec                     t           j                            t          j        d| z                       }d |D             }t          j                    }t          |          D ]\  }}|                    ||           t           j                            |          D ]-\  }}|                    t          j	        ||                     .|S )N)r   c                 4    g | ]\  }}t          ||          S  )range).0startends      Bshare/doc/networkx-3.1/examples/drawing/plot_multipartite_graph.py
<listcomp>z&multilayered_graph.<locals>.<listcomp>   s&    :::JE3eE3:::    layer)
nxutilspairwise	itertools
accumulateGraph	enumerateadd_nodes_fromadd_edges_fromproduct)subset_sizesextentslayersGir   layer1layer2s           r   multilayered_graphr(      s    h	 4TL5H I IJJG::':::F


Af%% ) )5	a(((((++F33 < <	*66::;;;;Hr   c                 :    g | ]\  }}t           |d                   S r   )subset_color)r   vdatas      r   r   r   $   s%    GGGDd7m	$GGGr   T)r,   r   )
subset_key)   r.   )figsizeF)
node_colorwith_labelsequal)__doc__r   matplotlib.pyplotpyplotpltnetworkxr   r!   r*   r(   r$   nodescolormultipartite_layoutposfiguredrawaxisshowr   r   r   <module>r@      s                  '''	 	 	   %GGAGGG4F4FGGGbQ7333 

6     35e 4 4 4 4     




r   