← К задачам
Тяжело · +6БагиМассивы

Помогите Пете починить код

Петя решает задачу на 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]
]
py
Курс по теме
Пройдите курс «Python с нуля» — по шагам, с проверкой
8 уроков · ~14 ч · теория, упражнения и экзамен с бейджем
Открыть курс →
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
Здесь появится результат. Напиши решение слева и нажми «Проверить код».
Для запуска тестов необходима авторизация.