← К задачам
Помогите Пете починить код
Петя решает задачу на CodeChick. Ему нужно написать функцию, которая принимает аргумент size — количество строк в матрице.
Функция должна вернуть матрицу с заданным количество строк, в которой диагонали заполнены 1, остальные ячейки — 0.
Петин код вы найдете на вкладке Код. Найдите и исправьте ошибки.
Примеры
help_bobby(1) ➞ [[1]]
help_bobby(3) ➞ [
[1, 0, 1],
[0, 1, 0],
[1, 0, 1]
]
help_bobby(4) ➞ [
[1, 0, 0, 1],
[0, 1, 1, 0],
[0, 1, 1, 0],
[1, 0, 0, 1]
]
def help_bobby(size):
array=[[[0]*size]*size]
for i in array:print(i)
row=0
for column in range(size):
array[column][row]=1
array[size-column][row]=1
row+=1
for i in array:print(i)
return array Для запуска тестов необходима авторизация.
