请创建一个两列的DataFrame数据,自定义一个lambda函数用来两列之和,并将最终的结果添加到新的列'sum_columns'当中
时间: 2024-10-19 14:14:14 浏览: 20
python中pandas.DataFrame对行与列求和及添加新行与列示例
当然可以,首先你需要创建一个包含两列的DataFrame,比如我们有两个列`column1`和`column2`。然后你可以使用`pandas`的`applymap()`或`lambda`函数计算两列的和。下面是创建这样一个DataFrame并添加新列的过程:
```python
import pandas as pd
# 创建一个简单的二维数组
data = {'column1': [1, 2, 3], 'column2': [4, 5, 6]}
# 创建 DataFrame
df = pd.DataFrame(data)
# 定义 lambda 函数,用于计算两列之和
sum_func = lambda x: x[0] + x[1]
# 新增一列 'sum_columns',存储两列之和
df['sum_columns'] = df[['column1', 'column2']].apply(sum_func, axis=1)
print(df)
```
在这个例子中,`axis=1`表示沿着行的方向(即每一行)应用`sum_func`。
运行这段代码后,你会得到这样的结果:
```
column1 column2 sum_columns
0 1 4 5
1 2 5 7
2 3 6 9
```
阅读全文