NumPy 乘积
乘积(Products)
要查找数组中元素的乘积,请使用 prod()
函数。
实例
求该数组中元素的乘积:
import numpy as np arr = np.array([1, 2, 3, 4]) x = np.prod(arr) print(x)
返回: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]