[Python] suma wyrazów z tablicy kwadratowej

[ct985]

Mam problem z napisaniem funkcji, która dla tablicy kwadratowej A zwraca tablicę kwadratową, która na pozycji (a, b) ma sumę wszystkich pozycji (c, d) z tablicy A, dla których 0 ≤ c ≤ a i 0 ≤ d ≤ b. Mój pomysł:

Kod: Zaznacz cały

def suma_iter(A):
	tablica=[]
	tab_wew=[]
	for k in range(len(A)):
		tab_wew.append(None)
	for k in range(len(A)):
		tablica.append(tab_wew)
	suma=0
	for i in range(len(A)):
		for j in range(len(A)):
			tablica[i][j]=suma+A[i][j]
			suma+=A[i][j]
	return tablica

[pfauel]

A ja mam taki pomysł

Kod: Zaznacz cały

def suma_iter(A):
    res = []
    for i_line, line in enumerate(A):
        res.append([])
        line_sum = 0
        for i_elem, elem in enumerate(line):
            line_sum += elem
            res[i_line].append(line_sum + (res[i_line-1][i_elem] if i_line > 0 else 0))
    return res

Mam nadzieję, że pomogłem. ()
pozdrawiam