pandas中cumsum() 函数的应用

cumsum() 函数用于计算 DataFrame 或 Series 中数值型数据的累积和。它将沿着指定的轴（行或列）对数据进行累积求和，并返回一个具有相同形状的 DataFrame 或 Series。

下面是一个示例，说明如何使用 cumsum() 函数：

import pandas as pd

# 创建一个 DataFrame
data = {
    'A': [1, 2, 3, 4, 5],
    'B': [4, 5, 6, 7, 8],
    'C': [7, 8, 9, 10, 11]
}
df = pd.DataFrame(data)

# 计算整个 DataFrame 的累积和
total_cumsum = df.cumsum()
print("Total cumulative sum of DataFrame:")
print(total_cumsum)

# 计算每列的累积和
column_cumsum = df.cumsum(axis=0)
print("\nColumn cumulative sum:")
print(column_cumsum)

# 计算每行的累积和
row_cumsum = df.cumsum(axis=1)
print("\nRow cumulative sum:")
print(row_cumsum)

输出结果：

Total cumulative sum of DataFrame:
    A   B   C
0   1   4   7
1   3   9  15
2   6  15  24
3  10  22  34
4  15  30  45

Column cumulative sum:
    A   B   C
0   1   4   7
1   3   9  15
2   6  15  24
3  10  22  34
4  15  30  45

Row cumulative sum:
    A   B   C
0   1   5  12
1   2   7  15
2   3   9  18
3   4  11  21
4   5  13  24

在这个示例中，我们首先创建了一个 DataFrame，并使用 cumsum() 函数计算了整个 DataFrame 的累积和、每列的累积和以及每行的累积和。可以通过指定 axis 参数来沿着行或列进行计算累积和，默认情况下是对列进行计算累积和。