Как переделать код в двумерный массив?

Question

[System04]

Есть код...

a = int(input())
k = 0
for i in range(a):
    for j in range(a):
        print(k, end=' ')
        k += 1
    print()

Как можно переделать его в двумерный массив (на список)?

Comments

[Vindicar]

В чём именно затруднение?
Не знаешь как создать список? Не знаешь как добавить элемент в список?

Answer 1

[Алан Гибизов]

Можно вот так:

def make_row(size):
    global k
    old_k = k
    k = old_k + size
    return [*range(old_k, k)]


if __name__ == '__main__':
    size = int(input('введите размерность массива:'))
    k = 0
    print([make_row(size) for i in range(size)])