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 的乘积
如果指定 axis=1
,NumPy 将返回每个数组的乘积。
实例
在以下数组中对第一个轴执行求和:
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]