Problem z macierzą

Question

Witam, mam zadanie do wykonania:http://prac.im.pwr.edu.pl/~giniew/doku.php?id=rok1920%3Azimowy%3Awip%3Azd&fbclid=IwAR3I9_qj0SPNfVubE5euv_6nO8bpsNkyg-2eu6zXXe3y-DSM-_tWeRFXPUM

i mam problem z napisaniem funkcji krok, ponieważ chciałbym ją wykonać w sposób taki, że za pomocą pętli będę sprawdzał każdą wartość utworzonej macierzy po kolei, ale nie wiem jak to napisać nie mając podanych jej wymiarów(niestety zadanie nakazuje stworzenie funkcji, której argumentem jest tylko plansza). Czy ktoś mógłby mi podpowiedzieć jak to zrobić?

Tutaj dwie poprzednie funkcje z zadania, które udało mi się napisać:

import numpy

def  nowa_plansza(H, W, pola=None):
    L = [[0]* H for i in range(W)]
    L = numpy.arange(H*W).reshape((H,W)) * 0
    for i,j in pola:
        L[i][j] = 1
    return L


def sasiedzi(plansza, i ,j):
    suma = 0
    
    if(plansza[i-1][j-1] == 1):
        suma += 1
            
    if(plansza[i][j-1] == 1):
        suma += 1
    
    if(plansza[i+1][j-1] == 1):
        suma += 1
    
    if(plansza[i-1][j] == 1):
        suma += 1
    
    if(plansza[i-1][j+1] == 1):
        suma += 1
    
    if(plansza[i][j+1] == 1):
        suma += 1
    
    if(plansza[i+1][j+1] == 1):
        suma += 1
    
    if(plansza[i-1][j-1] == 1):
        suma += 1
    return suma

 

Answer 1

Nie potrzebujesz takich rozmiarów. Możesz iterować bezpośrednio po macierzy
 

for x in L:
   # wiersze
   for y in x:
       # elementy wiersza

jak potrzebujesz indeksów użyj funkcji enumerate.

Comments

Dziękuję za odpowiedź, ale już sobie sam poradziłem z problemem. W dokumentacji znalazłem funkcje shape, dzięki której uzyskałem rozmiary.