Python NumPy prod方法用法及代碼示例

例子

基本用法

np.prod([1,2,3,4])



24

計算二維數組的乘積

考慮以下二維數組：

a = np.array([[1,2],[3,4]])
a



array([[1, 2],
       [3, 4]])

所有值

np.prod(a)



24

按列

np.prod(a, axis=0)



array([3, 8])

逐行

np.prod(a, axis=1)



array([ 2, 12])

指定輸出數組

a = np.zeros(2)
np.prod([[1,2],[3,4]], axis=1, out=a)   # row-wise product
a



array([ 2., 12.])

在這裏，我們將結果輸出到數組 a 。

指定初始值

np.prod([1,2,3], initial=10)



60

在這裏，由於我們設置了初始值 10，所以我們有 10*1*2*3 = 60 。

指定布爾掩碼

np.prod([4,5,6,7], where=[False, True, True, False])



30

此處，僅第二個和第三個值包含在乘積的計算中。