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

此处，仅第二个和第三个值包含在乘积的计算中。