Calificación:
  • 0 voto(s) - 0 Media
  • 1
  • 2
  • 3
  • 4
  • 5
Ayuda con suma independiente de filas de matriz
#1
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.
Responder
#2
Hola, ¿cómo estás? El siguiente código te puede servir:

  1. matrix = [[9, 0, 9], [36, 1, 1]]
  2. result = []
  3. for sub in matrix:
  4.    result.append(sum(sub))
  5. 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:

  1. matrix = [[9, 0, 9], [36, 1, 1]]
  2. result = [sum(sub) for sub in matrix]
  3. print(result)


Saludos!
Responder


Salto de foro:


Usuarios navegando en este tema: 1 invitado(s)