Suma de elementos correspondientes de la lista en Python

Este intercambio describirá cómo sumar los elementos correspondientes de múltiples listas en Python, siempre que la longitud de cada lista sea la misma. Por ejemplo: a = [1,2,3], b = [2,3,4], c = [3,4,5], sume los elementos correspondientes de a, b, c, la salida debe ser [6 , 9,12]
  
Método 1:
  Resolver directamente: De acuerdo con el principio de sumar los elementos correspondientes, primero se puede definir una función.

def list_add(a,b):
    c = []
    for i in range(len(a)):
        c.append(a[i]+b[i])
    return c

if __name__ == '__main__':
    a = [1,2,3]
    b = [2,3,4]
    c = [3,4,5]
    print(list_add(list_add(a,b),c))

Método 2:
  utilice el módulo numpy para resolver el problema.

import numpy as np
a = np.array([1,2,3])
b = np.array([2,3,4])
c = np.array([3,4,5])
print(a+b+c)

Cabe señalar que el tipo después de a + b + c es numpy.ndarray
Método 3:
  use la función sum () del módulo numpy para resolver.

import numpy as np
a = [1,2,3]
b = [2,3,4]
c = [3,4,5]
print(np.sum([a,b,c], axis = 0))

El parámetro del eje representa la suma vertical.


Este intercambio termina aquí, todos son bienvenidos a comunicarse ~~

**Nota:** Ahora abrí una cuenta oficial de WeChat: rastreador y algoritmo de Python (ID de WeChat: easy_web_scrape), bienvenidos a todos a prestar atención ~~

Supongo que te gusta

Origin blog.csdn.net/jclian91/article/details/78118805
Recomendado
Clasificación