NumPy 求和
求和
求和與加法有什麼區別?
加法是兩個引數之間的運算,而求和是對 n 個元素進行的運算。
示例
將 arr1 中的值與 arr2 中的值相加
import numpy as np
arr1 = np.array([1, 2, 3])
arr2 = np.array([1, 2, 3])
newarr = np.add(arr1, arr2)
print(newarr)
自己動手試一試 »
返回: [2 4 6]
示例
對 arr1 和 arr2 中的值進行求和
import numpy as np
arr1 = np.array([1, 2, 3])
arr2 = np.array([1, 2, 3])
newarr = np.sum([arr1, arr2])
print(newarr)
自己動手試一試 »
返回: 12
沿軸求和
如果指定 axis=1
,NumPy 會對每個陣列中的數字進行求和。
示例
沿第一軸對以下陣列進行求和
import numpy as np
arr1 = np.array([1, 2, 3])
arr2 = np.array([1, 2, 3])
newarr = np.sum([arr1, arr2], axis=1)
print(newarr)
自己動手試一試 »
返回: [6 6]
累積和
累積和意味著對陣列中的元素進行部分相加。
例如,[1, 2, 3, 4] 的部分和將是 [1, 1+2, 1+2+3, 1+2+3+4] = [1, 3, 6, 10]。
使用 cumsum()
函式執行部分和。
示例
對以下陣列執行累積求和
import numpy as np
arr = np.array([1, 2, 3])
newarr = np.cumsum(arr)
print(newarr)
自己動手試一試 »
返回: [1 3 6]