Как присвоить значение 2d массиву в цикле for

Question

Я хочу присвоить значение 2d массиву для l oop
Это мой код

num = 0
n = 3
arr = [[0] * n] * n

for i in range(n):
    for j in range(n):
        arr[i][j] = num
        num +=1

Ожидаемый вывод:

[0, 1, 2]
[3, 4, 5]
[6, 7, 8]

Но фактический вывод is

[6, 7, 8]
[6, 7, 8]
[6, 7, 8]

Есть ли способ исправить это?

Answer 1

arr = [[0] * n] * n

Создает n копий списка. Таким образом, когда вы вносите изменения в один список, все остальные также меняются. Вы можете изменить его на что-то вроде этого:

arr = [[0 for j in n] for i in n]