Ayuda con suma independiente de filas de matriz

[pegomezca]

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.

[Francisco]

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!