Python numpy cumprod用法及代碼示例

本文簡要介紹 python 語言中 numpy.cumprod 的用法。

用法:

numpy.cumprod(a, axis=None, dtype=None, out=None)

返回沿給定軸的元素的累積乘積。

參數：

a： array_like

輸入數組。

axis： 整數，可選

計算累積乘積的軸。默認情況下，輸入是展平的。

dtype： dtype，可選

返回數組的類型，以及元素相乘的累加器的類型。如果未指定 dtype，則默認為 a 的 dtype，除非 a 具有精度小於默認平台整數的整數 dtype。在這種情況下，將使用默認平台整數。

out： ndarray，可選

用於放置結果的替代輸出數組。它必須具有與預期輸出相同的形狀和緩衝區長度，但如果需要，將轉換結果值的類型。

返回：

cumprod： ndarray

除非指定了 out，否則將返回一個保存結果的新數組，在這種情況下，將返回對 out 的引用。

注意：

使用整數類型時算術是模塊化的，溢出時不會引發錯誤。

例子：

>>> a = np.array([1,2,3])
>>> np.cumprod(a) # intermediate results 1, 1*2
...               # total product 1*2*3 = 6
array([1, 2, 6])
>>> a = np.array([[1, 2, 3], [4, 5, 6]])
>>> np.cumprod(a, dtype=float) # specify type of output
array([   1.,    2.,    6.,   24.,  120.,  720.])

a 的每一列(即行上)的累積乘積：

>>> np.cumprod(a, axis=0)
array([[ 1,  2,  3],
       [ 4, 10, 18]])

a 的每一行(即列上)的累積乘積：

>>> np.cumprod(a,axis=1)
array([[  1,   2,   6],
       [  4,  20, 120]])