当前位置：首页 > article >正文

【五分钟】熟练使用numpy.cumsum()函数（干货！！！）

article 2024/9/20 9:44:36

引言

numpy.cumsum()函数用于计算输入数组的累积和。当输入是多维数组时，numpy.cumsum()函数可以沿着指定轴计算累积和。

计算一维数组的累计和

代码如下：

# 计算一维数组的累计和
tmp_array = np.ones((4,), dtype=np.uint8)  # [1, 1, 1, 1]
print("输入:", tmp_array)

# 方案1 ———— numpy数组的cumsum方法
array_cumsum = tmp_array.cumsum()
print("方案1输出", array_cumsum)

# 方案2 ———— numpy的cumsum函数
array_cumsum = np.cumsum(a=tmp_array)
print("方案2输出:", array_cumsum)

运行结果：
在这里插入图片描述
可以看到，当输入是长度为4的全1数组时，方案一和方案二的计算结果完全一致，即输出数组的第i个元素计算的是输入数组的前i项和(累计和)

计算二维数组的累计和（沿着纵轴）

代码如下：

# 沿着纵轴计算二维数组的累计和
tmp_array = np.ones((4,4), dtype=np.uint8)
print("输入:", tmp_array)

# 方案1 ———— numpy数组的cumsum方法
array_cumsum = tmp_array.cumsum(axis=0) # 0代表沿着纵轴
print("方案1输出", array_cumsum)

# 方案2 ———— numpy的cumsum函数
array_cumsum = np.cumsum(a=tmp_array, axis=0)
print("方案2输出", array_cumsum)

运行结果：
在这里插入图片描述
可以看到，当输入是形状为(4,4)的全1数组时，方案一和方案二都是沿着输入数组的纵轴去计算累积和。

计算二维数组的累计和（沿着横轴）

# 沿着横轴计算二维数组的累计和
tmp_array = np.ones((4,4), dtype=np.uint8)
print("输入:", tmp_array)

# 方案1 ———— numpy数组的cumsum方法
array_cumsum = tmp_array.cumsum(axis=1) # 1代表沿着横轴
print("方案1输出", array_cumsum)

# 方案2 ———— numpy的cumsum函数
array_cumsum = np.cumsum(a=tmp_array, axis=1)
print("方案2输出", array_cumsum)

运行结果
在这里插入图片描述
可以看到，当输入是形状为(4,4)的全1数组时，方案一和方案二都是沿着输入数组的横轴去计算累积和。

计算二维数组的累积和(不指定轴)

代码如下：

tmp_array = np.ones((4,4), dtype=np.uint8)
print("输入:\n", tmp_array)

# 方案1 ———— numpy数组的cumsum方法
array_cumsum = tmp_array.cumsum() # 不指定轴
print("方案1输出:\n", array_cumsum)

# 方案2 ———— numpy的cumsum函数
array_cumsum = np.cumsum(a=tmp_array)
print("方案2输出:\n", array_cumsum)