Итерация по словарю для создания списка элементов внутри другого списка

Question

У меня есть словарь под названием «Флот нескольких кораблей».Каждый корабль в словаре длиннее 1 (состоит как минимум из двух наборов координат, например [[3, 1], [3, 2]] и [[2, 3], [2, 4], [2, 5]].

. С помощью приведенного ниже кода мне удается получить координаты всех кораблей,но почему-то я не могу понять, как составить список флотов, где каждый элемент внутри представляет собой список списков с координатами. Для приведенного выше примера мне нужен такой результат:

[[[3, 1], [3, 2]],
 [[2, 3], [2, 4], [2, 5]]]

Я представляю другой *Требуется цикл 1008 *, но мне не удается правильно его интегрировать.

Проблемный код находится в функции place_ship ()

num_cols = 5  # board size direction x
num_rows = 4  # board size direction y
empty_field = "~"

fleet = {
"Submarine": [3, "S"],
"Patrol Boat": [2, "P"]
}

def place_ship():
    # place ship based on orientation
    list_ship_coordinates = []
    if ori == "v":
        for i in range(fleet[ship][0]):
            board[x + i][y] = fleet[ship][1]
            part_of_ship = [x + i + 1, y + 1]
            list_ship_coordinates.append(part_of_ship)
        print(list_ship_coordinates)
    elif ori == "h":
        for i in range(fleet[ship][0]):
            board[x][y + i] = fleet[ship][1]
            part_of_ship = [x + 1, y + i + 1]
            list_ship_coordinates.append(part_of_ship)
        print(list_ship_coordinates)
    return board


def validate():
    if ori == "v" and x + fleet[ship][0] > num_rows:
        return False
    elif ori == "h" and y + fleet[ship][0] > num_cols:
        return False
    else:
        if ori == "v":
            for i in range(fleet[ship][0]):
                if board[x + i][y] != empty_field:
                    return False
        elif ori == "h":
            for i in range(fleet[ship][0]):
                if board[x][y + i] != empty_field:
                    return False
    return True


for ship in fleet:
    valid = False
    while not valid:
        x = randint(0, num_rows - 1)
        y = randint(0, num_cols - 1)
        o = randint(0, 1)
        if o == 0:
            ori = "v"
        else:
            ori = "h"
        valid = validate()
    board = place_ship()

Answer 1

Извините, что не спросил четко.

Оказывается, новые строки 71 и 78 в функции place_ship решили мою проблему.

from random import randint

num_cols = 5  # board size direction x
num_rows = 4  # board size direction y
empty_field = "~"
cheat_mode = 1  # 1 = Yes; 0 = No
line1 = "   "
line2 = "------"

# Type and quantity of ships in fleet
fleet = {
    "Aircraft Carrier": [5, "A"],
    "Battleship": [4, "B"],
    "Destroyer": [3, "D"],
    "Submarine": [3, "S"],
    "Patrol Boat": [2, "P"]
}

all_ship_parts = []
for ship in fleet:
    all_ship_parts1 = fleet[ship][0]
    all_ship_parts.append(all_ship_parts1)

all_ship_parts = sum(all_ship_parts)
num_of_turns = all_ship_parts + 1

board = []
# Create a blank board.
for each_row in range(num_rows):
    board_col = []
    for each_col in range(num_cols):
        board_col.append(empty_field)
    board.append(board_col)


def print_board():
    print("    ", end="")
    for i in range(0, num_cols):
        print(" " + str(i + 1) + " ", end=' ')
        print(" ", end=' ')
    print("")
    print(line1 + line2 * num_cols)

    for i in range(0, num_rows):
        if i < 9:
            print(str(i + 1) + " | ", end=' ')
        else:
            print(str(i + 1) + "| ", end=' ')
        for j in range(0, num_cols):
            if board[i][j] == empty_field:
                print(" ", end=' ')
            else:
                print(board[i][j], end=' ')
            print(" | ", end=' ')
        print("")
        print(line1 + line2 * num_cols)


# print_board()
total = []


def place_ship():
    # place ship based on orientation
    list_ship_coordinates = []
    if ori == "v":
        for i in range(fleet[ship][0]):
            board[x + i][y] = fleet[ship][1]
            part_of_ship = [x + i + 1, y + 1]
            list_ship_coordinates.append(part_of_ship)
        total.append(list_ship_coordinates)
        print(list_ship_coordinates)
    elif ori == "h":
        for i in range(fleet[ship][0]):
            board[x][y + i] = fleet[ship][1]
            part_of_ship = [x + 1, y + i + 1]
            list_ship_coordinates.append(part_of_ship)
        total.append(list_ship_coordinates)
        print(list_ship_coordinates)

    return board


def validate():
    if ori == "v" and x + fleet[ship][0] > num_rows:
        return False
    elif ori == "h" and y + fleet[ship][0] > num_cols:
        return False
    else:
        if ori == "v":
            for i in range(fleet[ship][0]):
                if board[x + i][y] != empty_field:
                    return False
        elif ori == "h":
            for i in range(fleet[ship][0]):
                if board[x][y + i] != empty_field:
                    return False
    return True


for ship in fleet:
    valid = False
    while not valid:
        x = randint(0, num_rows - 1)
        y = randint(0, num_cols - 1)
        o = randint(0, 1)
        if o == 0:
            ori = "v"
        else:
            ori = "h"
        valid = validate()
    board = place_ship()

print("-----------")
print(total)
print("-----------")
print(len(total))
print(sum(len(x) for x in total))
print("-----------")
print_board()
# print(board[0][2])
# print(board[1][2])
# print(board[2][2])
# print(board[3][2])