Python之list对应元素求和

发布时间:2019-09-16 07:25:22编辑:auto阅读(1844)

    本次分享将讲述如何在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参数表示纵向求和。


    本次分享到此结束,欢迎大家交流~~

    **注意:**本人现已开通微信公众号: Python爬虫及算法(微信号为:easy_web_scrape), 欢迎大家关注哦~~

关键字

上一篇: python 统计文件行数

下一篇: 安装Python(Mac)