本次分享將講述如何在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 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(微信號為:python_math)以及輕松學會Python爬蟲(微信號為:easy_web_scrape), 歡迎大家關注哦~~