Pandas 数据框:计算总和并添加到新列和行
Pandas 数据框:计算总和并添加到新列和行
在 Pandas 数据框中,您可以使用 iloc 和 sum 方法来计算总和并将其添加到新的列和行。
目标:
- 计算数据框中除第一列外的所有列的总和,并将结果添加到名为 'total' 的新列中。
- 计算数据框中所有行的总和,并将结果添加到名为 'total' 的新行中。
代码示例:
import pandas as pd
# 创建一个示例数据框
df_pivot = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
# 计算除第一列外的所有列的总和,并添加到新列
df_pivot['total'] = df_pivot.iloc[:, 1:].sum(axis=1)
# 计算所有行的总和,并添加到新行
df_pivot.loc['total'] = df_pivot.sum()
print(df_pivot)
解释:
df_pivot.iloc[:, 1:]选择了数据框的所有行,以及从第二列 (索引 1) 开始的所有列。sum(axis=1)沿着行进行求和,生成每行的总和。df_pivot.loc['total']将新行命名为 'total'。df_pivot.sum()计算所有列的总和,生成每一列的总和。
注意:
根据您的数据和需求,您可以修改代码以选择不同的列和行进行求和。
本指南展示了一种在 Pandas 数据框中计算总和并添加到新列和行的方法。通过 iloc 和 sum 方法,您可以灵活地处理数据并根据您的需求进行定制。
原文地址: https://www.cveoy.top/t/topic/f58 著作权归作者所有。请勿转载和采集!