使用Python中的国际象棋库从 PGN 文件中提取数据

国际象棋是一种广受欢迎的棋类游戏，许多人喜欢在网上观看国际象棋比赛，并且保存PAN格式（PGN）文件以记录比赛的详细信息。使用Python中的国际象棋库，可以轻松地从PGN文件中提取数据以便分析和使用。

安装Python的国际象棋库

首先需安装 python-chess 库。可以通过运行以下命令来安装它：

pip install python-chess

读取PGN文件

PGN（Portable Game Notation）是一种用于记录国际象棋游戏的文件格式。Python的国际象棋库可以很容易地读取PGN文件，如下所示：

import chess.pgn

pgn = open("game.pgn")
game = chess.pgn.read_game(pgn)

在上面的代码中，首先通过 import 语句导入了 chess.pgn 模块。接着使用 open() 函数打开了一个PGN文件，然后调用了 read_game() 函数以将文件中保存的国际象棋游戏读取到一个变量中（在上面的例子中，变量名为 game）。请注意，以上的代码只会读取PGN文件中的第一场比赛，如果文件中存在多场比赛，需要使用循环读取。

解析PGN文件

在使用 chess.pgn 库成功读取游戏数据后，我们可以再次使用该库来解析数据并进行各种操作，例如生成游戏列表或从游戏中提取数据等。如下所示：

import chess.pgn

pgn = open("game.pgn")

while True:
    game = chess.pgn.read_game(pgn)
    if game is None:
        break
    print(game.headers["White"], "vs", game.headers["Black"], "\nResult:", game.headers["Result"], "\n")

在上述代码中， while 循环检查是否存在其他游戏，如果存在则将其读入变量 game 中。由于此处循环遍历了文件中的所有比赛，因此可以对每一场比赛进行操作。在这种情况下，代码解析了每一场比赛的头和标签，以及双方玩家的姓名和比赛结果。输出的结果通常类似于以下格式：

Max Euwe vs Mikhail Botvinnik 
Result: 0-1

提取走棋法

从一个正确格式的 PGN 文件中提取走棋法可以通过下面的代码进行实现：

import chess.pgn

pgn = open("game.pgn")

while True:
    game = chess.pgn.read_game(pgn)
    if game is None:
        break

    board = game.board()
    print(f"Game: {game.headers['Event']}")
    print(f"White: {game.headers['White']}")
    print(f"Black: {game.headers['Black']}")
    for move in game.mainline_moves():
        board.push(move)
        print(board.san(move))

上述代码解析PGN文件，然后循环遍历文件中的每个游戏。在循环中，为棋盘初始化并以棋谱形式转换每个走法。最后的输出结果类似于：

Game: Game of the century
White: Donald Byrne
Black: Robert James Fischer
Nf3
Nf6
g3
g6
Bg2
Bg7
O-O
d5
c4
dxc4
Qxd8
Rxd8
Na3
c3
Bf4
Nc6
Nb5
Bg4
Nc7
Rac8
Nfe5
Nxe5
Bxe4
Bxd1
Bxc6
bxc6
Nxe7
Bxb2
Rab1
Bxc1
Nxc8
Rd1
Rxd1
Bxd1
Be5
Bxa7
Kg7
a4
Kf6
a5
Ke6
a6
Kd7
Bb8
h5
Kf3
Ke6
g4
hxg4
hxg4
f5
gxf5
gxf5
e3
fxe3
fxe3
Be5
Ke4
Kd3
Kc3
Kb4
Ka3
Kb4
Bd4
Kc3
Kb4
Ka3
Kb4
Bxe3
f4
Bg2
Bf1
f3
Bxf3
Kc3
Bxc4
Kb4
Be7
Ka5
Bd6
Kb6
Bc5
Ka5
Ka7
Bb7

总结

使用 python-chess 库可以方便地从 PGN 文件中提取数据。可以解析头和标签字以及走棋法，以便更深入地分析国际象棋游戏数据。