python 将apply自定义函数处理结果合并到原数据集
时间: 2023-03-26 12:03:16 浏览: 113
合并数据集.py
可以使用 pandas 库中的 apply 方法来实现。具体步骤如下:
1. 定义自定义函数,对数据进行处理。
2. 使用 apply 方法将自定义函数应用到数据集上。
3. 将处理结果合并到原数据集中。
示例代码如下:
```python
import pandas as pd
# 定义自定义函数
def process_data(row):
# 对数据进行处理
return row['col1'] + row['col2']
# 创建数据集
data = {'col1': [1, 2, 3], 'col2': [4, 5, 6]}
df = pd.DataFrame(data)
# 使用 apply 方法将自定义函数应用到数据集上
df['result'] = df.apply(process_data, axis=1)
# 输出结果
print(df)
```
运行结果如下:
```
col1 col2 result
1 4 5
1 2 5 7
2 3 6 9
```
阅读全文