python 五子棋-文字版（上）

import random

row = 10
col = 10

def new_board():
    return [[" "]*10 for i in range(row)]

def display_board(board):
    for i in range(row):
        print(f"{'':-^39}")
        for j in range(col):
            print(board[i][j],end=" | ")
        print()
    print(f"{'':-^39}")


def pieces():
    is_go_first = input("你想要先走吗? (y/n): ").lower()
    if is_go_first == "y":
        human, computer = "X", "O"
    else:
        human, computer = "O", "X"
    return human, computer

def legal_moves(board):
    moves = []
    for i in range(row):
        for j in range(col):
            if board[i][j] == " ":
                moves.append((i,j))
    return moves

def computer_move(board):
    moves = legal_moves(board)
    pos = random.choice(moves)
    return pos


def win(board):
    # 行判断
    for i in range(row):
        for j in range(col-4):
            if board[i][j] == board[i][j+1] == board[i][j+2] == board[i][j+3] == board[i][j+4] != " ":
                return board[i][j]

    # 列判断
    for j in range(col):
        for i in range(row-4):
            if board[i][j] == board[i+1][j] == board[i+2][j] == board[i+3][j] == board[i+4][j] != " ":
                return board[i][j]

    # 左高右低判断
    for i in range(row-4):
        for j in range(col-4):
            if board[i][j] == board[i+1][j+1] == board[i+2][j+2] == board[i+3][j+3] == board[i+4][j+4] != " ":
                return board[i][j]

    # 左低右高判断
    for i in range(4,row):
        for j in range(col-4):
            if board[i][j] == board[i-1][j+1] == board[i-2][j+2] == board[i-3][j+3] == board[i-4][j+4]!=" ":
                return board[i][j]

    # 判断是否下完所有位置
    for i in range(row):
        for j in range(col):
            if board[i][j]==" ":
                return None
    else:
        return "平局"

def ask_move_pos(board):
    while True:
        pos = input("你的输入位置？(x,y):")
        str_x, str_y = pos.split(",")
        x, y = int(str_x), int(str_y)
        if board[x][y] == " ":
            return x, y

def next_turn(turn):
    if turn == "X":
        return "O"
    else:
        return "X"

board = new_board()
display_board(board)
human,computer = pieces()
turn = "X"
x,y=0,0
while not win:
    if turn ==human:
        x, y = ask_move_pos(board)
        board[x][y]=turn
    else:
        x,y = computer_move(board)
        board[x][y]=turn
    display_board(board)
    turn = next_turn(turn)
    
win = win(board)
print(win)