本次分享将讲述如何在Python中对多个list的对应元素求和,前提是每个list的长度一样。比如:a=[1,2,3], b=[2,3,4], c=[3,4,5], 对a,b,c的对应元素求和,输出应为[6,9,12].
方法一:
直接求解,按照对应元素相加的原则,可先定义一个函数。
1
2
3
4
5
6
7
8
9
10
11
|
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)) |
方法二:
利用numpy模块求解。
1
2
3
4
5
|
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) |
需要注意的是,a+b+c后的类型为numpy.ndarray.
方法三:
利用numpy模块的sum()函数进行求解。
1
2
3
4
5
|
import numpy as np a = [ 1 , 2 , 3 ] b = [ 2 , 3 , 4 ] c = [ 3 , 4 , 5 ] print (np. sum ([a,b,c], axis = 0 )) |
其中的axis参数表示纵向求和。
以上这篇Python之list对应元素求和的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/jclian91/article/details/78118805