Foros Python

Buenos días, a ver si podéis echarme una mano.
Tengo la siguiente matriz: [[9, 0, 9], [36, 1, 1]] y necesito obtener un vector con el resultado de sumar los elementos de cada submatriz, es decir, necesito que me salga esto: [18,38].
Me tiene loco, y no soy capaz de localizar la solución.
Muchas gracias de antemano.

Hola, ¿cómo estás? El siguiente código te puede servir:

Código:
matrix = [[9, 0, 9], [36, 1, 1]]

result = []

for sub in matrix:

   result.append(sum(sub))

print(result)

Básicamente recorres cada elemento de la matriz y sumas los elementos de la submatriz vía la función sum().

O bien una solución más pythonica usando comprensión de listas:

Código:
matrix = [[9, 0, 9], [36, 1, 1]]

result = [sum(sub) for sub in matrix]

print(result)

Saludos!

pegomezca

Francisco