Vh,dZgdZddlZddlmZmZedddd Zed dd Zejd dd  ddZ eddejddd  ddZ y)a ******************** Bipartite Edge Lists ******************** Read and write NetworkX graphs as bipartite edge lists. Format ------ You can read or write three formats of edge lists with these functions. Node pairs with no data:: 1 2 Python dictionary as data:: 1 2 {'weight':7, 'color':'green'} Arbitrary data:: 1 2 7 green For each edge (u, v) the node u is assigned to part 0 and the node v to part 1. )generate_edgelistwrite_edgelistparse_edgelist read_edgelistN)not_implemented_for open_filewb)modeTcrt|||D]'}|dz }|j|j|)y)a-Write a bipartite graph as a list of edges. Parameters ---------- G : Graph A NetworkX bipartite graph path : file or string File or filename to write. If a file is provided, it must be opened in 'wb' mode. Filenames ending in .gz or .bz2 will be compressed. comments : string, optional The character used to indicate the start of a comment delimiter : string, optional The string used to separate values. The default is whitespace. data : bool or list, optional If False write no edge data. If True write a string representation of the edge data dictionary.. If a list (or other iterable) is provided, write the keys specified in the list. encoding: string, optional Specify which encoding to use when writing file. Examples -------- >>> G = nx.path_graph(4) >>> G.add_nodes_from([0, 2], bipartite=0) >>> G.add_nodes_from([1, 3], bipartite=1) >>> nx.write_edgelist(G, "test.edgelist") >>> fh = open("test.edgelist_open", "wb") >>> nx.write_edgelist(G, fh) >>> nx.write_edgelist(G, "test.edgelist.gz") >>> nx.write_edgelist(G, "test.edgelist_nodata.gz", data=False) >>> G = nx.Graph() >>> G.add_edge(1, 2, weight=7, color="red") >>> nx.write_edgelist(G, "test.edgelist_bigger_nodata", data=False) >>> nx.write_edgelist(G, "test.edgelist_color", data=["color"]) >>> nx.write_edgelist(G, "test.edgelist_color_weight", data=["color", "weight"]) See Also -------- write_edgelist generate_edgelist  N)rwriteencode)Gpathcomments delimiterdataencodinglines V/home/dcms/DCMS/lib/python3.12/site-packages/networkx/algorithms/bipartite/edgelist.pyrr s;Z"!Y5*   4;;x()*directedc#6K |jjDcgc]\}}|ddk(s|}}}|dus|durA|D];}|j ||D]#}|j t t|%=y|D]^}|j |dD]F\}} || g} |jfd|D|j t t|H`ycc}}w#t$r}td|d}~wwxYw#t$rYTwxYww) aGenerate a single line of the bipartite graph G in edge list format. Parameters ---------- G : NetworkX graph The graph is assumed to have node attribute `part` set to 0,1 representing the two graph parts delimiter : string, optional Separator for node labels data : bool or list of keys If False generate no edge data. If True use a dictionary representation of edge data. If a list of keys use a list of data values corresponding to the keys. Returns ------- lines : string Lines of data in adjlist format. Examples -------- >>> from networkx.algorithms import bipartite >>> G = nx.path_graph(4) >>> G.add_nodes_from([0, 2], bipartite=0) >>> G.add_nodes_from([1, 3], bipartite=1) >>> G[1][2]["weight"] = 3 >>> G[2][3]["capacity"] = 12 >>> for line in bipartite.generate_edgelist(G, data=False): ... print(line) 0 1 2 1 2 3 >>> for line in bipartite.generate_edgelist(G): ... print(line) 0 1 {} 2 1 {'weight': 3} 2 3 {'capacity': 12} >>> for line in bipartite.generate_edgelist(G, data=["weight"]): ... print(line) 0 1 2 1 3 2 3 bipartiterz"Missing node attribute `bipartite`NTF)rc3(K|] }| ywN).0kds r z$generate_edgelist..s3!3s) nodesitems BaseExceptionAttributeErroredgesjoinmapstrextendKeyError) rrrnr!part0erredgeuvs ` rrrRs+bL ww}}Ftq!!K.A2EFF t|tu} 5A- 5nnSd^44 5 5 5A771470 51a1vKK3d33 nnSd^44  5 5G LABKL saDC-C'C'C-A,D'D &D'C-- D6 DDD DDDDbipartite_parse_edgelist)namegraphs returns_graphc ddlm}tjd|}|D]c}|j |} | dk\r|d| }t |s+|j dj|} t | dkrZ| jd} | jd} | } | || } || } t | dk(s|d uri}n|d ur" t|d j| }n^t | t |k7rtd| d|di}t|| D]$\\}}} ||}|j||i&|j!| d|j!| d|j"| | fi|f|S#t$r}td| d| d|d |d}~wwxYw#t$r}td | d|d}~wwxYw#t$r}td|d|d|d |d}~wwxYw)aParse lines of an edge list representation of a bipartite graph. Parameters ---------- lines : list or iterator of strings Input data in edgelist format comments : string, optional Marker for comment lines delimiter : string, optional Separator for node labels create_using: NetworkX graph container, optional Use given NetworkX graph for holding nodes or edges. nodetype : Python type, optional Convert nodes to this type. data : bool or list of (label,type) tuples If False generate no edge data or if True use a dictionary representation of edge data or a list tuples specifying dictionary key names and types for edge data. Returns ------- G: NetworkX Graph The bipartite graph corresponding to lines Examples -------- Edgelist with no data: >>> from networkx.algorithms import bipartite >>> lines = ["1 2", "2 3", "3 4"] >>> G = bipartite.parse_edgelist(lines, nodetype=int) >>> sorted(G.nodes()) [1, 2, 3, 4] >>> sorted(G.nodes(data=True)) [(1, {'bipartite': 0}), (2, {'bipartite': 0}), (3, {'bipartite': 0}), (4, {'bipartite': 1})] >>> sorted(G.edges()) [(1, 2), (2, 3), (3, 4)] Edgelist with data in Python dictionary representation: >>> lines = ["1 2 {'weight':3}", "2 3 {'weight':27}", "3 4 {'weight':3.0}"] >>> G = bipartite.parse_edgelist(lines, nodetype=int) >>> sorted(G.nodes()) [1, 2, 3, 4] >>> sorted(G.edges(data=True)) [(1, 2, {'weight': 3}), (2, 3, {'weight': 27}), (3, 4, {'weight': 3.0})] Edgelist with data in a list: >>> lines = ["1 2 3", "2 3 27", "3 4 3.0"] >>> G = bipartite.parse_edgelist(lines, nodetype=int, data=(("weight", float),)) >>> sorted(G.nodes()) [1, 2, 3, 4] >>> sorted(G.edges(data=True)) [(1, 2, {'weight': 3.0}), (2, 3, {'weight': 27.0}), (3, 4, {'weight': 3.0})] See Also -------- r) literal_evalNr zFailed to convert nodes ,z to type .FT zFailed to convert edge data (z) to dictionary.z Edge data z and data_keys z are not the same lengthzFailed to convert z data )rr )astr8nx empty_graphfindlenrstripsplitpopr% TypeErrordictr( IndexErrorzipupdateadd_nodeadd_edge)linesrr create_usingnodetyperr8rrpsr1r2r!r/edgedataedge_key edge_type edge_values rrrsP~! q,'A3% IIh  68D4y  KK  # #I . q6A:  EE!H EE!H    QKQK q6Q;$%-H T\  SXXa[ 9:1vT"  ?4&8PQH58q\ 81%9z!*:!6J : 67 8 1 " 1 " 1a$8$g3%h HG! .qc1#YxjJ ! 3A36FG %#,XJf%,i {!=sH F F/,G F,F''F,/ G 8GG  G4G//G4rbbipartite_read_edgelistc<fd|D}t||||||S)aRead a bipartite graph from a list of edges. Parameters ---------- path : file or string File or filename to read. If a file is provided, it must be opened in 'rb' mode. Filenames ending in .gz or .bz2 will be decompressed. comments : string, optional The character used to indicate the start of a comment. delimiter : string, optional The string used to separate values. The default is whitespace. create_using : Graph container, optional, Use specified container to build graph. The default is networkx.Graph, an undirected graph. nodetype : int, float, str, Python type, optional Convert node data from strings to specified type data : bool or list of (label,type) tuples Tuples specifying dictionary key names and types for edge data edgetype : int, float, str, Python type, optional OBSOLETE Convert edge data from strings to specified type and use as 'weight' encoding: string, optional Specify which encoding to use when reading file. Returns ------- G : graph A networkx Graph or other type specified with create_using Examples -------- >>> from networkx.algorithms import bipartite >>> G = nx.path_graph(4) >>> G.add_nodes_from([0, 2], bipartite=0) >>> G.add_nodes_from([1, 3], bipartite=1) >>> bipartite.write_edgelist(G, "test.edgelist") >>> G = bipartite.read_edgelist("test.edgelist") >>> fh = open("test.edgelist", "rb") >>> G = bipartite.read_edgelist(fh) >>> fh.close() >>> G = bipartite.read_edgelist("test.edgelist", nodetype=int) Edgelist with data in a list: >>> textline = "1 2 3" >>> fh = open("test.edgelist", "w") >>> d = fh.write(textline) >>> fh.close() >>> G = bipartite.read_edgelist( ... "test.edgelist", nodetype=int, data=(("weight", float),) ... ) >>> list(G) [1, 2] >>> list(G.edges(data=True)) [(1, 2, {'weight': 3.0})] See parse_edgelist() for more examples of formatting. See Also -------- parse_edgelist Notes ----- Since nodes must be hashable, the function nodetype must return hashable types (e.g. int, float, str, frozenset - or tuples of those, etc.) c3@K|]}|jywr)decode)rrrs rr"z read_edgelist..`s 4tT[[ " 4s)rrrMrNr)r) rrrrMrNredgetyperrLs ` rrrs/b 5t 4E  !   r)#r<Tutf-8)r<T)r[NNNT)r[NNNTNr\) __doc____all__networkxr>networkx.utilsrrrr _dispatchablerrrrrrbs2 U9 14.*.*bZ @5!@5F1$dSPTu Tu p 140TR   WSWr