python numpy用法_Python numpy.cumprod()用法及代码示例

当我们要计算给定轴上数组元素的累积乘积时，将使用numpy.cumprod()函数。

用法： numpy.cumprod(arr, axis=None, dtype=None, out=None)

参数：

arr :[数组]包含需要累积乘积的数字的数组。如果arr不是数组，则尝试进行转换。

axis :计算累积乘积的轴。默认值是计算展平数组的乘积。

dtype :返回数组的类型，以及与元素相乘的累加器的类型。如果未指定dtype，则默认为arr的dtype，除非arr的整数dtype的精度小于默认平台整数的精度。在这种情况下，将使用默认平台整数。

out :[ndarray，可选]将结果存储到的位置。

->如果提供，则必须具有广播输入的形状。

->如果未提供或没有，则返回新分配的数组。

Return :除非指定out，否则将返回保存结果的新数组，在这种情况下将返回该数组。

代码1：工作

# Python program explaining

# numpy.cumprod() function

import numpy as geek

in_num = 10

print ("Input  number:", in_num)

out_prod = geek.cumprod(in_num)

print ("cumulative product of input number:", out_prod)

输出：

Input number: 10

cumulative product of input number: [10]

代码2：

# Python program explaining

# numpy.cumprod() function

import numpy as geek

in_arr = geek.array([[2, 4, 6], [1, 3, 5]])

print ("Input array:", in_arr)

out_prod = geek.cumprod(in_arr)

print ("cumulative product of array elements:", out_prod)

输出：

Input array: [[2 4 6]

[1 3 5]]

cumulative product of array elements: [ 2 8 48 48 144 720]

代码3：

# Python program explaining

# numpy.cumprod() function

import numpy as geek

in_arr = geek.array([[2, 4, 6], [1, 3, 5]])

print ("Input array:", in_arr)

out_prod = geek.cumprod(in_arr, axis = 1)

print ("cumulative product of array elements taking axis 1:", out_prod)

输出：

Input array: [[2 4 6]

[1 3 5]]

cumulative product of array elements taking axis 1: [[ 2 8 48]

[ 1 3 15]]