es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Suma de matrices en Numpy.

Tengo 4 arreglos de numpy distintos cada uno con 40 valores, y quiero sumar todos los elementos en la posición 0 de los 4 arreglos juntos y almacenarlos en otro arreglo en la posición 0. Todos los elementos en la posición 1 los quiero almacenar en la posición 1 y así sucesivamente para los 40 elementos de los 4 arreglos. ¿Cómo debería hacerlo en Python?

Tags:  , ,

Answer

  1. Avatar for davy.ai

    Para lograr esto, puedes utilizar la función numpy.sum() junto con el parámetro axis. Aquí tienes un ejemplo de código:

    import numpy as np
    
    # creando 4 arreglos con 40 valores cada uno
    a1 = np.random.rand(40)
    a2 = np.random.rand(40)
    a3 = np.random.rand(40)
    a4 = np.random.rand(40)
    
    # creando un arreglo vacío de tamaño 40
    result = np.empty(40)
    
    # sumando todos los elementos en posiciones correspondientes
    result[0] = np.sum([a1[0], a2[0], a3[0], a4[0]])
    result[1] = np.sum([a1[1], a2[1], a3[1], a4[1]])
    # y así sucesivamente para los 40 elementos
    
    # Alternativamente, puedes utilizar un ciclo for para evitar codificar manualmente
    for i in range(40):
        result[i] = np.sum([a1[i], a2[i], a3[i], a4[i]])
    
    # Imprimiendo los resultados
    print(result)
    

    Esto debería darte la salida deseada, donde result[0] tendrá la suma de todos los elementos en la posición 0 de los cuatro arreglos, y así sucesivamente para todos los demás índices.

Comments are closed.