如何将多列的总和添加到数据框中的另一列 - Python

在数据分析中，将多列的总和添加到数据框中的另一列是一项常见任务。在本文中，我们将介绍如何使用Python Pandas库来实现这一目标。

示例数据

首先，我们需要创建一个示例数据集。在本文中，我们将使用以下代码来创建一个具有三列数据的数据框：

import pandas as pd

# 创建数据集
data = {'A': [1, 2, 3],
        'B': [4, 5, 6],
        'C': [7, 8, 9]}

df = pd.DataFrame(data)

print(df)

输出如下所示：

   A  B  C
0  1  4  7
1  2  5  8
2  3  6  9

在本例中，我们有三列数据：A、B和C。

计算总和

接下来，我们需要计算每一行中A、B和C列的总和。在本例中，我们可以使用以下代码来计算每一行的总和：

# 计算每一行的总和
row_sums = df.sum(axis=1)

print(row_sums)

这将输出每一行的总和：

0    12
1    15
2    18
dtype: int64

将总和添加到另一列

现在，我们已经计算了每一行的总和，下一步是将它们添加到数据框中的另一列。在本例中，我们将创建一个名为“D”的新列，用于存储每一行的总和。

# 添加新列
df['D'] = row_sums

print(df)

这将输出一个包含新" D "列的修改后的数据框：

   A  B  C   D
0  1  4  7  12
1  2  5  8  15
2  3  6  9  18

结论

通过这种方式，我们可以将多列的总和添加到数据框中的另一列。这对于数据分析和可视化等任务非常有用。在Python中，使用Pandas库可以轻松完成这项任务。希望这篇文章对您有所帮助！