o ˷e@sdZddlmZddlZgdZdHddZeZe de de d e Z d d ee DZeed krFe d ed ddejddDZe deede de dejddD]/\ZZZeddkrededZeD]\ZZededed7Zqed7Ze eqieeZ dd e DZeedZejddD]:\Z ZZ!e!d"dZ#e#dd kree d!7<qe#dd"kree d#7<eed#7<qeed!7<qd$d e DZ$e%e Z&e&d%d&7<e&d'd(7<e&d)d*7<ej'd+d,\Z(Z)ej*e e&d-ed.d/ej+e e&e$d0d1d2d3d4d5d6Z,ej-e e&d7e,d8d9d3d:d7d;Z.e)/d<e)j0d?d@dAdBe)j1e.dCe)j0d?dDdEdBe)j1e.dCe)2d@dFe(3e4dGe5dS)Ia  ============= Chess Masters ============= An example of the MultiDiGraph class. The function `chess_pgn_graph` reads a collection of chess matches stored in the specified PGN file (PGN ="Portable Game Notation"). Here the (compressed) default file:: chess_masters_WCC.pgn.bz2 contains all 685 World Chess Championship matches from 1886--1985. (data from http://chessproblem.my-free-games.com/chess/games/Download-PGN.php) The `chess_pgn_graph()` function returns a `MultiDiGraph` with multiple edges. Each node is the last name of a chess master. Each edge is directed from white to black and contains selected game info. The key statement in `chess_pgn_graph` below is:: G.add_edge(white, black, game_info) where `game_info` is a `dict` describing each game. N)EventDateResultECOSitechess_masters_WCC.pgn.bz2c Csddl}t}i}||}dd|D}Wdn1s!wY|D]5}|drE|dddd\}}|d |t|<q(|r]|d } |d } |j | | fi|i}q(|S) zRead chess games in pgn format in pgn_file. Filenames ending in .bz2 will be uncompressed. Return the MultiDiGraph of players connected by a chess game. Edges contain game data in a dict. rNcSsg|] }|dqS)z )decoderstrip).0liner |/var/www/ideatree/venv/lib/python3.10/site-packages/../../../share/doc/networkx-2.8.6/examples/drawing/plot_chess_masters.py 2z#chess_pgn_graph..[ "WhiteBlack) bz2nx MultiDiGraphBZ2File startswithsplitstripstrpopadd_edge) pgn_filerGgamedatafilelinesr tagvaluewhiteblackr r r chess_pgn_graph$s"     r*zLoaded z chess games between z players cCsg|]}t|qSr )Hsubgraph)r cr r r rJsrrz/Note the disconnected component consisting of: cCsh|] \}}}|dqS)rr )r r(r) game_infor r r Orr/T)dataz From a total of z different openings,z-the following games used the Sicilian openingz5with the Najdorff 7...Qb6 "Poisoned Pawn" variation. rB97z vs  z z: cCs g|] \}}tt||qSr )lenr" get_edge_data)r uvr r r r`s gr-1g?z1/2g?cCsg|]}t|dqS)2)wins)r r6r r r rmszReshevsky, Samuel H)皙?gzBotvinnik, Mikhail M)gQ?gQzSmyslov, Vassily V)r;gQ) r<)figsizeg333333?m)alphawidth edge_colorz#210070g?) node_size node_colorr?kr(gffffff?)ecfcr?) font_sizebbox Helveticabold)fontnamecolor fontweightfontsizez+World Chess Championship Games: 1886 - 1985rrMg?g?zedge width = # games playedcenter)horizontalalignment transformfontdictgQ?znode size = # games wonr;off)r)6__doc__matplotlib.pyplotpyplotpltnetworkxr game_detailsr*r"printnumber_of_edgesnumber_of_nodes to_undirectedr+connected_componentsGccr3nodesedgesopeningsr(r)r.summaryitemsrDr6Graph edgewidthdictfromkeysr:r5drrPnodesizekamada_kawai_layoutpossubplotsfigaxdraw_networkx_edgesdraw_networkx_nodes label_optionsdraw_networkx_labelsfont set_titletext transAxesmargins tight_layoutaxisshowr r r r s