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 ~~