Помогите с list comprehention задачка.

С помощью list comprehention создайте двумерную матрицу (список списков) размера 4×4, в которой находятся квадраты последовательных целых чисел, начиная с 0. Сначала заполняется первый столбец, затем второй и т.д. Формат вывода представлен в ячейке ниже.

# пример вывода для 3х3
# [0, 9, 36]
# [1, 16, 49]
# [4, 25, 64]

Делал вот такое.
[0, 1, 4, 9]
[0, 1, 4, 9]
[0, 1, 4, 9]
[0, 1, 4, 9]

Голову уже сломал. Помогите плиз.

stud_55 сказал(а):

Вот пример:

Python:

answer = [[(i + j * 4) ** 2 for j in range(4)] for i in range(4)]
for i in answer:
    print(i)

Нажмите, чтобы раскрыть...

Спасибо большое. Благодарю.