python dataframe 多列相加

原创

mob649e81693c66 2023-12-28 07:33:42 ©著作权

文章标签 多列 python Python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e81693c66的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python Pandas DataFrame 多列相加

在数据分析和处理中，我们经常需要对数据框中的多列进行相加操作。Python中的Pandas库提供了DataFrame对象来处理和操作表格数据。本文将介绍如何使用Pandas的DataFrame对象对多列进行相加，并提供代码示例。

1. 创建DataFrame

在介绍如何对多列相加之前，我们先创建一个简单的DataFrame对象作为示例数据。我们将使用Pandas的DataFrame()函数，并传入一个字典作为参数，其中键表示列名，值表示该列的数据。

import pandas as pd

data = {'col1': [1, 2, 3, 4, 5],
        'col2': [6, 7, 8, 9, 10],
        'col3': [11, 12, 13, 14, 15]}
df = pd.DataFrame(data)

以上代码创建了一个DataFrame对象df，包含了三列（col1、col2、col3）和五行数据。

2. 多列相加

接下来，我们将介绍如何对DataFrame中的多列进行相加操作。Pandas中的DataFrame对象支持对列进行加法运算，通过指定列名即可进行相加。

df['col_sum'] = df['col1'] + df['col2'] + df['col3']

以上代码将col1、col2和col3三列相加，并将结果赋值给新的列col_sum。可以看到，我们通过直接使用列名进行相加操作，非常方便。

3. 结果展示

我们可以通过打印DataFrame对象来查看添加了新列col_sum后的结果。

print(df)

输出结果如下：

   col1  col2  col3  col_sum
0     1     6    11       18
1     2     7    12       21
2     3     8    13       24
3     4     9    14       27
4     5    10    15       30

可以看到，新的列col_sum显示了col1、col2和col3三列相加的结果。

4. 流程图

下面是对以上操作的流程图，使用mermaid语法标识：

flowchart TD
  A[创建DataFrame] --> B[多列相加]
  B --> C[结果展示]

5. 完整代码

下面是完整的代码示例：

import pandas as pd

# 创建DataFrame
data = {'col1': [1, 2, 3, 4, 5],
        'col2': [6, 7, 8, 9, 10],
        'col3': [11, 12, 13, 14, 15]}
df = pd.DataFrame(data)

# 多列相加
df['col_sum'] = df['col1'] + df['col2'] + df['col3']

# 结果展示
print(df)