o
    ˷e                     @   s   d d Z dS )c           
      C   s   ddl }g }i }d}t| jddD ]%\}}||d v r7|d | }||vr.|||< |d7 }|||| g q||}|dd t| dd	 d
D }	||	fS )a  Get and return information of labels from the input graph

    Parameters
    ----------
    G : Network X graph
    label_name : string
        Name of the target label

    Returns
    ----------
    labels : numpy array, shape = [n_labeled_samples, 2]
        Array of pairs of labeled node ID and label ID
    label_dict : numpy array, shape = [n_classes]
        Array of labels
        i-th element contains the label corresponding label ID `i`
        NT)data   c                 S   s   g | ]\}}|qS  r   ).0label_r   r   d/var/www/ideatree/venv/lib/python3.10/site-packages/networkx/algorithms/node_classification/utils.py
<listcomp>    s    z#_get_label_info.<locals>.<listcomp>c                 S   s   | d S )Nr   r   )xr   r   r   <lambda>    s    z!_get_label_info.<locals>.<lambda>)key)numpy	enumeratenodesappendarraysorteditems)
G
label_namenplabelslabel_to_idlidinr   
label_dictr   r   r   _get_label_info   s"   
r   N)r   r   r   r   r   <module>   s    