Aby wpisać dane do macierza (dwa wymiary) należy zastosować dwie zagnieżdżone pętle:
for x in macierz:
for y in x:
przypisanie
Aby uzyskać numer kolumny można skorzystać z funkcji enumerate:
macierz = [[_ for _ in range(3)] for _ in range(3)] # deklaracja
for x, rzad in enumerate(macierz): # rzad to rząd macierza, x to numer rzędu
for kolumna, _ in enumerate(rzad): # tutaj kolumna to numer kolumny, _ nic nie robi
macierz[x][kolumna] = kolumna
Wynik:
print(macierz)
# [[0, 1, 2], [0, 1, 2], [0, 1, 2]]