首页 > 编程 > Python > 正文

Python之list对应元素求和的方法

2020-02-15 22:04:17
字体:
来源:转载
供稿:网友

本次分享将讲述如何在Python中对多个list的对应元素求和,前提是每个list的长度一样。比如:a=[1,2,3], b=[2,3,4], c=[3,4,5], 对a,b,c的对应元素求和,输出应为[6,9,12].   

方法一:

直接求解,按照对应元素相加的原则,可先定义一个函数。

def list_add(a,b): c = [] for i in range(len(a)): c.append(a[i]+b[i]) return cif __name__ == '__main__': a = [1,2,3] b = [2,3,4] c = [3,4,5] print(list_add(list_add(a,b),c))

方法二:

利用numpy模块求解。

import numpy as npa = 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()函数进行求解。

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

其中的axis参数表示纵向求和。

以上这篇Python之list对应元素求和的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持武林站长站。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表