Python中的战舰游戏-随机放置战舰

import random

def place_ship(board, ship_length):
    while True:
        # 随机选择起始行和列
        start_row = random.randint(0, len(board) - 1)
        start_col = random.randint(0, len(board[0]) - 1)
        
        # 确定战舰的方向（水平或垂直）
        direction = random.choice(['horizontal', 'vertical'])
        
        if direction == 'horizontal':
            # 检查水平方向是否足够放置战舰
            if start_col + ship_length <= len(board[0]):
                for i in range(ship_length):
                    board[start_row][start_col + i] = 'S'
                break
        else:
            # 检查垂直方向是否足够放置战舰
            if start_row + ship_length <= len(board):
                for i in range(ship_length):
                    board[start_row + i][start_col] = 'S'
                break
    
    return board

# 初始化5x5的游戏板
game_board = [['O' for _ in range(5)] for _ in range(5)]

# 放置战舰
place_ship(game_board, 3)

# 打印游戏板
for row in game_board:
    print(' '.join(row))