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.])

這裏，. 表示數字是浮點數。