U md@sdZddlmZddlZdddddgZdJd d ZeZe d e d e d e Z ddee DZeedkre dedddejddDZe deede de dejddD]`\ZZZeddkrededZeD] \ZZededed7Zqed7Ze eqeeZ dde DZeed ZejddD]z\Z ZZ!e!d"d!Z#e#dd"kree d#7<n@e#dd$kree d%7<eed%7<need#7<qrd&de 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/ed0d1ej+e e&e$d2d3d4d5d6d7d8Z,ej-e e&d9e,d:d;d5de.d?e.d@<e)j0dAdBdCdDe)j1e.dEe)j0dAdFdGdDe)j1e.dEe)2dBdHe(3e4dIe5dS)Ka  ============= 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. NEventDateResultECOZSitechess_masters_WCC.pgn.bz2c Csddl}t}i}||}dd|D}W5QRX|D]f}|drv|dddd\}}|d |t|<q<|r<|d } |d } |j | | f|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 |/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/../../../share/doc/networkx-3.1/examples/drawing/plot_chess_masters.py 2sz#chess_pgn_graph..[ "WhiteZBlack) bz2nxZ MultiDiGraphBZ2File startswithsplitstripstrpopZadd_edge) Zpgn_filerGZgameZdatafilelinesr tagvaluewhiteblackr r r chess_pgn_graph$s     r"zLoaded z chess games between z players cCsg|]}t|qSr )HZsubgraph)r cr r r r Jsr rz/Note the disconnected component consisting of: cCsh|]\}}}|dqS)rr )r r r! game_infor r r Osr&T)dataz From a total of z different openings,z-the following games used the Sicilian openingz5with the Najdorff 7...Qb6 "Poisoned Pawn" variation. ZB97z vs  z z: cCs g|]\}}tt||qSr )lenrZ get_edge_data)r uvr r r r `sg-1g?z1/2g?cCsg|]}t|dqS)2)wins)r r+r r r r mszReshevsky, Samuel H)皙?gzBotvinnik, Mikhail M)gQ?gQzSmyslov, Vassily V)r0gQ) r1)Zfigsizeg333333?m)alphawidthZ edge_colorz#210070g?)Z node_sizeZ node_colorr3kr gffffff?)ecfcr3)Z font_sizeZbboxZ Helveticabold)ZfontnamecolorZ fontweightZfontsizez+World Chess Championship Games: 1886 - 1985rr:g?g?zedge width = # games playedcenter)ZhorizontalalignmentZ transformZfontdictgQ?znode size = # games wonr0off)r)6__doc__Zmatplotlib.pyplotZpyplotZpltZnetworkxrZ game_detailsr"rprintZnumber_of_edgesZnumber_of_nodesZ to_undirectedr#Zconnected_componentsZGccr)ZnodesedgesZopeningsr r!r%summaryitemsr5r+ZGraphZ edgewidthdictfromkeysr/r*drr;ZnodesizeZkamada_kawai_layoutposZsubplotsZfigZaxZdraw_networkx_edgesZdraw_networkx_nodesZ label_optionsZdraw_networkx_labelsZfont set_titletextZ transAxesZmarginsZ tight_layoutZaxisshowr r r r s