Изменение списков в матричном алгоритме Python

Вам по-прежнему потребуются вложенные циклы for тем или иным образом, поскольку вы выполняете итерацию по списку списков, но вы можете немного упростить логику, используя понимание списков.

def solver(matrix):
     mx = [[v if v else 'x' for v in row] for row in matrix]
     mxx = [[v1 if v2 else 'x' for v1, v2 in zip(row1, row2)] for row1, row2 in zip(mx[1:], matrix)]
     return mx[:1] + mxx

Сначала я перебираю матрицу и заменяю 0 на "x" в новой матрице mx.

mx = [[v if v else 'x' for v in row] for row in matrix]

Это просто понимание вложенного списка, где мы работаем с каждым элементом в строке, для каждой строки в каждой матрице. ... if ... else ... — это просто классический тернарный оператор. Если выполняется v (в нашем случае не ноль), то оно оценивается как значение до «если», в противном случае оно оценивается как значение после «иначе» — в данном случае 'x'.

Затем я повторяю процесс, но смещаю строки на единицу, чтобы проверить, является ли вышеприведенный элемент теперь "x".

mxx = [[v1 if v2 else 'x' for v1, v2 in zip(row1, row2)] for row1, row2 in zip(mx[1:], matrix)]

Тут есть что ломать. Начнем с «снаружи» и пройдем внутрь.

... for row1, row2 in zip(mx[1:], matrix)

Это упаковывает новую матрицу со смещением на единицу (с использованием обозначения среза [1:]) с исходной матрицей. Таким образом, он возвращает итерируемый объект, функционально эквивалентный следующему списку:

[(mx_row1, matrix_row0), (mx_row2, matrix_row1), (mx_row3, matrix_row2), ...]

Это позволяет нам одновременно извлекать заданную строку и строку над ней, как row1 и row2. Потом другая половина-

[v1 if v2 else 'x' for v1, v2 in zip(row1, row2)]

- повторяет аналогичный процесс для каждого элемента в строке, а не для строки в матрице. Мы не смещаем элементы ни в одной строке, как мы смещаем строки матрицы mx, но в остальном логика идентична. Затем мы снова сравниваем с нашим тернарным оператором, чтобы увидеть, является ли вышеуказанный элемент 0, и если да, то оцениваем его как 'x'. Мы могли бы легко изменить это, чтобы сравнивать каждый элемент каждой строки от mx до 'x' вместо matrix до 0, но я решил отразить первое понимание списка.

Когда у меня есть эта новая матрица mxx, я просто добавляю первую строку mx, потому что мы фактически пропускаем эту строку, когда смещаем наше сравнение. Результатом является матрица, в которой все 0s и элементы ниже заменены на "x"s.

Согласно пояснению в комментариях, если вы хотите пометить "x", если какой-либо из вышеперечисленных элементов имеет значение 0, а не только тот, который непосредственно выше, вы можете сделать это, взяв фрагмент этого столбца матрицу и используя встроенную функцию all(), чтобы увидеть, являются ли они 0. Пересмотренный код ниже

def solver(matrix):
    return [[v if all(col) else 'x' for v, col in zip(row, zip(*matrix[:idx]))] for idx, row in enumerate(matrix, 1)]