o
    ˷e                     @   s   d Z ddlZddlmZ ddlZg dZg dZdd Z	e	e Z
dd e
jd	d
D Zeje
ddZejdd eje
eedd ed e  dS )z=
===================
Multipartite Layout
===================
    N)   r            r   r   r   )goldvioletr   r   r   	limegreenr   
darkorangec                  G   sx   t jtd|  }dd |D }t  }t|D ]\}}|j||d qt j|D ]\}}|t	|| q,|S )N)r   c                 S   s   g | ]	\}}t ||qS  )range).0startendr
   r
   /var/www/ideatree/venv/lib/python3.10/site-packages/../../../share/doc/networkx-2.8.6/examples/drawing/plot_multipartite_graph.py
<listcomp>   s    z&multilayered_graph.<locals>.<listcomp>layer)
nxutilspairwise	itertools
accumulateGraph	enumerateadd_nodes_fromadd_edges_fromproduct)subset_sizesextentslayersGir   layer1layer2r
   r
   r   multilayered_graph   s   r$   c                 C   s   g | ]
\}}t |d   qS r   )subset_color)r   vdatar
   r
   r   r   $   s    r   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
   r   <module>   s    
