sum()
函数用于计算 DataFrame 或 Series 中数值型数据的总和。它可以对整个 DataFrame 或 Series 进行求和,也可以沿着指定的轴(行或列)进行求和。
下面是一个示例,说明如何使用 sum()
函数:
import pandas as pd
# 创建一个 DataFrame
data = {
'A': [1, 2, 3],
'B': [4, 5, 6],
'C': [7, 8, 9]
}
df = pd.DataFrame(data)
# 计算整个 DataFrame 的总和
total_sum = df.sum()
print("Total sum of DataFrame:")
print(total_sum)
# 计算每列的总和
column_sum = df.sum(axis=0)
print("\nColumn sum:")
print(column_sum)
# 计算每行的总和
row_sum = df.sum(axis=1)
print("\nRow sum:")
print(row_sum)
输出结果:
Total sum of DataFrame:
A 6
B 15
C 24
dtype: int64
Column sum:
A 6
B 15
C 24
dtype: int64
Row sum:
0 12
1 15
2 18
dtype: int64
在这个示例中,我们首先创建了一个 DataFrame,并使用 sum()
函数计算了整个 DataFrame 的总和、每列的总和以及每行的总和。可以通过指定 axis
参数来沿着行或列进行求和,默认情况下是对列进行求和。