python如何对list求和

2,107次阅读
没有评论

python如何对list求和

如何在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 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模块求解。

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()函数进行求解。

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参数表示纵向求和。

神龙|纯净稳定代理IP免费测试>>>>>>>>天启|企业级代理IP免费测试>>>>>>>>IPIPGO|全球住宅代理IP免费测试

相关文章:

版权声明:wuyou2021-06-28发表,共计569字。
新手QQ群:570568346,欢迎进群讨论 Python51学习