python想获得一列值为其他列相加的列怎么实现
时间: 2024-05-11 07:16:13 浏览: 96
python中利用numpy.array()实现俩个数值列表的对应相加方法
5星 · 资源好评率100%
可以使用 Pandas 库中的 Dataframe,利用 apply 函数和 lambda 表达式来实现。
具体操作步骤如下:
1. 读取数据到 Pandas Dataframe 中。
2. 定义一个函数,接收每一行的数据,对需要相加的列进行加法运算,返回相加的结果。
3. 使用 Dataframe 的 apply 函数,将刚刚定义的函数应用到数据的每一行上,得到一个包含新列的 Dataframe。
示例代码如下:
```
import pandas as pd
# 读取数据到 Dataframe
df = pd.read_csv('data.csv')
# 定义一个函数,实现相加的操作
def sum_of_columns(row):
return row['column_1'] + row['column_2'] + row['column_3']
# 对每一行应用函数,得到新列
df['sum_column'] = df.apply(lambda row: sum_of_columns(row), axis=1)
```
以上代码将 Dataframe 中的 column_1、column_2、column_3 三列相加,得到新列 sum_column,可以使用该列进行后续的操作。
阅读全文