在python中将dataframe的第一列和第二列经过函数def1处理后赋值给第三列
时间: 2024-02-11 20:09:32 浏览: 85
python pandas.DataFrame.loc函数使用详解
5星 · 资源好评率100%
可以使用以下代码将DataFrame的第一列和第二列经过函数`def1`处理后赋值给第三列:
```python
import pandas as pd
# 定义一个函数def1,用于处理DataFrame的第一列和第二列
def def1(x, y):
return x + y
# 创建一个DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [0, 0, 0]})
# 将第一列和第二列经过函数def1处理后赋值给第三列
df['C'] = df.apply(lambda row: def1(row['A'], row['B']), axis=1)
```
在上面的代码中,`def1`是我们定义的一个处理函数,用于将输入值相加。`df.apply(lambda row: def1(row['A'], row['B']), axis=1)`会将DataFrame的每一行作为输入,经过`def1`函数处理后返回一个新的Series对象,该对象包含处理后的值。最后,将这个Series对象赋值给DataFrame的第三列。
阅读全文