cumsum函数python|极客教程

cumsum函数python

在Python中，cumsum()是一个用于计算累积和的函数，它可以对一个数组或序列中的元素进行累积求和操作。在本文中，我们将详细介绍cumsum()函数的用法和示例。

语法

cumsum()函数的语法如下：

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

参数说明：

a：需要进行累积求和操作的数组或序列。
axis：沿着哪个轴进行累积求和操作，默认是对数组所有元素进行求和。
dtype：返回数组的数据类型，默认是输入数组的数据类型。
out：可选参数，用于指定输出的数组。

示例

假设我们有一个包含1到10的列表，我们可以使用cumsum()函数来计算其累积和。下面是一个简单的示例：

import numpy as np

# 创建一个包含1到10的列表
arr = np.arange(1, 11)

# 计算列表的累积和
cumulative_sum = np.cumsum(arr)

print(cumulative_sum)

运行以上代码将输出以下结果：

[ 1  3  6 10 15 21 28 36 45 55]

以上结果表示，原始数组 [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 的累积和依次为 [1, 3, 6, 10, 15, 21, 28, 36, 45, 55]。

指定axis参数

cumsum()函数还可以指定axis参数来沿着指定的轴进行累积求和操作。例如，对于一个二维数组，我们可以分别对行和列进行累积求和。下面是一个示例：

import numpy as np

# 创建一个二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 沿着行的方向计算累积和
row_cumsum = np.cumsum(arr, axis=1)

# 沿着列的方向计算累积和
col_cumsum = np.cumsum(arr, axis=0)

print("沿着行的方向计算累积和：\n", row_cumsum)
print("沿着列的方向计算累积和：\n", col_cumsum)

运行以上代码将输出以下结果：

沿着行的方向计算累积和：
 [[ 1  3  6]
 [ 4  9 15]
 [ 7 15 24]]
沿着列的方向计算累积和：
 [[ 1  2  3]
 [ 5  7  9]
 [12 15 18]]

从以上结果可以看出，对于二维数组 [[1, 2, 3], [4, 5, 6], [7, 8, 9]]，沿着行的方向计算累积和得到的数组是 [[1, 3, 6], [4, 9, 15], [7, 15, 24]]，而沿着列的方向计算累积和得到的数组是 [[1, 2, 3], [5, 7, 9], [12, 15, 18]]。

指定dtype参数

cumsum()函数还可以指定dtype参数来设置返回数组的数据类型。如果不指定dtype参数，默认会使用输入数组的数据类型。下面是一个示例：

import numpy as np

# 创建一个包含浮点数的数组
arr_float = np.array([1.1, 2.2, 3.3])

# 指定返回数组的数据类型为整数
cumsum_int = np.cumsum(arr_float, dtype=np.int)

print(cumsum_int)

运行以上代码将输出以下结果：

[1 3 6]

从以上结果可以看出，对于包含浮点数 [1.1, 2.2, 3.3] 的数组，指定返回数组的数据类型为整数后，累积和数组为 [1, 3, 6]。

指定out参数

cumsum()函数还可以指定out参数来指定输出的数组，这样可以节省内存空间。下面是一个示例：

import numpy as np

# 创建一个包含1到5的列表
arr = np.arange(1, 6)

# 创建一个空的数组作为输出结果
output_arr = np.empty(5)

# 计算累积和，并指定输出结果数组
np.cumsum(arr, out=output_arr)

print(output_arr)

运行以上代码将输出以下结果：