Python NumPy cumprod方法用法及代码示例

例子

一维数组

计算一维数组的累积乘积：

x = np.array([1,2,3])
np.cumprod(x)



array([1, 2, 6])

在这里，我们正在执行以下计算：

[0] 1 = 1
[1] 1 * 2 = 2
[2] 1 * 2 * 3 = 6

二维数组

假设我们有以下二维数组：

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



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

所有值

要计算数组中所有值的累积乘积：

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



array([ 1,  2,  6, 24])

按列

要按列计算累积乘积，请设置 axis=0 ：

x = np.array([[1,2], [3,4]])
np.cumprod(x, axis=0)



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

逐行

要按行计算累积乘积，请设置 axis=1 ：

x = np.array([[1,2], [3,4]])
np.cumprod(x, axis=1)



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

指定数据类型

要获取数据类型 float 的数组：

x = np.array([1,2,3])
np.cumprod(x, dtype=float)



array([1., 2., 6.])

这里，. 表示数字是浮点数。