U
    ª9%eá  ã                   @   s°   d Z ddlZddlmZ ddlZddddddddgZdddddd	d	d
g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é   é   é   é   ZgoldÚvioletZ	limegreenZ
darkorangec                  G   sx   t j t d|  ¡¡}dd„ |D ƒ}t  ¡ }t|ƒD ]\}}|j||d q4t j |¡D ]\}}| t 	||¡¡ qX|S )N)r   c                 S   s   g | ]\}}t ||ƒ‘qS © )Úrange)Ú.0ÚstartÚendr   r   ú‡/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/../../../share/doc/networkx-3.1/examples/drawing/plot_multipartite_graph.pyÚ
<listcomp>   s     z&multilayered_graph.<locals>.<listcomp>©Úlayer)
ÚnxÚutilsÚpairwiseÚ	itertoolsÚ
accumulateZGraphÚ	enumerateZadd_nodes_fromZadd_edges_fromÚproduct)Úsubset_sizesZextentsZlayersÚGÚir   Zlayer1Z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   )Z
subset_key)é   r   )ZfigsizeF)Z
node_colorZwith_labelsÚequal)Ú__doc__r   Zmatplotlib.pyplotZpyplotZpltZnetworkxr   r   r   r   r   ZnodesÚcolorZmultipartite_layoutÚposZfigureZdrawZaxisÚshowr   r   r   r   Ú<module>   s*   ø
