o h' @sjdZddlZddlmZddlmZddgZedd ejdd dZd d Z edd ejd dZ dS)z8Algorithms to calculate reciprocity in a directed graph.N) NetworkXError)not_implemented_for reciprocityoverall_reciprocity undirected multigraphcCsL|durt|S||vrtt||d}|durtd|Stt||S)afCompute the reciprocity in a directed graph. The reciprocity of a directed graph is defined as the ratio of the number of edges pointing in both directions to the total number of edges in the graph. Formally, $r = |{(u,v) \in G|(v,u) \in G}| / |{(u,v) \in G}|$. The reciprocity of a single node u is defined similarly, it is the ratio of the number of edges in both directions to the total number of edges attached to node u. Parameters ---------- G : graph A networkx directed graph nodes : container of nodes, optional (default=whole graph) Compute reciprocity for nodes in this container. Returns ------- out : dictionary Reciprocity keyed by node label. Notes ----- The reciprocity is not defined for isolated nodes. In such cases this function will return None. NzNot defined for isolated nodes.)rnext_reciprocity_iterrdict)Gnodesrrs/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/networkx/algorithms/reciprocity.pyr s!c cs|||}|D]3}t||}t||}||@}t|t|}|dkr.|dfVqdt||}||fVqdS)z*Return an iterator of (node, reciprocity).rNr) nbunch_iterset predecessors successorslen) r rnnodepredsuccoverlapn_totalrrrrr =s   r cCs4|}||d}|dkrtd||S)zCompute the reciprocity for the whole graph. See the doc of reciprocity for the definition. Parameters ---------- G : graph A networkx graph rrzNot defined for empty graphs)number_of_edges to_undirectedr)r n_all_edgen_overlap_edgerrrrOs  )N) __doc__networkxnxrutilsr__all__ _dispatchablerr rrrrrs  0