Python之list对应元素求和的方法
本次分享将讲述如何在Python中对多个list的对应元素求和,前提是每个list的长度一样。比如:a=[1,2,3],b=[2,3,4],c=[3,4,5],对a,b,c的对应元素求和,输出应为[6,9,12].
方法一:
直接求解,按照对应元素相加的原则,可先定义一个函数。
deflist_add(a,b): c=[] foriinrange(len(a)): c.append(a[i]+b[i]) returnc if__name__=='__main__': a=[1,2,3] b=[2,3,4] c=[3,4,5] print(list_add(list_add(a,b),c))
方法二:
利用numpy模块求解。
importnumpyasnp 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()函数进行求解。
importnumpyasnp a=[1,2,3] b=[2,3,4] c=[3,4,5] print(np.sum([a,b,c],axis=0))
其中的axis参数表示纵向求和。
以上这篇Python之list对应元素求和的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。