1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
def solution(matrix): row = [None] * 4 column = [None] * 4 for i in range(4): for j in range(4): if matrix[i][j] == 0 : row[i] = 1 column[j] = 1 for i in range(4): for j in range(4): if row[i] or column[j] == 1 : matrix[i][j] = 0 return matrix print(solution([[1, 2, 3, 4], [2, 4, 0, 0], [4, 0, 8, 9],[1, 3, 5, 1]])) |