NumPy 乘積
產品
要計算陣列中元素的乘積,請使用 prod()
函式。
返回: 24
因為 1*2*3*4 = 24
示例
計算兩個陣列中元素的乘積
import numpy as np
arr1 = np.array([1, 2, 3, 4])
arr2 = np.array([5, 6, 7, 8])
x = np.prod([arr1, arr2])
print(x)
自己動手試一試 »
返回: 40320
因為 1*2*3*4*5*6*7*8 = 40320
沿軸的乘積
如果指定 axis=1
,NumPy 將返回每個陣列的乘積。
示例
在以下陣列的第 1 個軸上執行求和
import numpy as np
arr1 = np.array([1, 2, 3, 4])
arr2 = np.array([5, 6, 7, 8])
newarr = np.prod([arr1, arr2], axis=1)
print(newarr)
自己動手試一試 »
返回: [24 1680]
累積乘積
累積乘積意味著部分地進行乘積。
例如,[1, 2, 3, 4] 的部分乘積是 [1, 1*2, 1*2*3, 1*2*3*4] = [1, 2, 6, 24]
使用 cumprod()
函式執行部分求和。
示例
計算以下陣列中所有元素的累積乘積
import numpy as np
arr = np.array([5, 6, 7, 8])
newarr = np.cumprod(arr)
print(newarr)
自己動手試一試 »
返回: [5 30 210 1680]