函数处理dataframe并返回
时间: 2023-12-31 10:04:35 浏览: 16
好的,我可以给你一个例子来说明如何编写一个函数来处理 DataFrame 并返回结果。
下面是一个简单的示例函数,它接收一个 DataFrame 作为输入,并返回一个处理后的 DataFrame:
```python
import pandas as pd
def process_dataframe(df):
# 对 DataFrame 进行处理,例如添加一列数据
df['new_col'] = df['col1'] + df['col2']
# 返回处理后的 DataFrame
return df
```
在这个示例函数中,我们首先对传入的 DataFrame 进行了处理,例如添加了一个新的列来存储两个列的和。然后,我们返回了处理后的 DataFrame。
你可以在调用函数时向其传递任何 DataFrame,例如:
```python
# 创建一个 DataFrame
df = pd.DataFrame({'col1': [1, 2, 3], 'col2': [4, 5, 6]})
# 调用处理函数
result_df = process_dataframe(df)
# 输出处理后的 DataFrame
print(result_df)
```
这个函数将输出以下内容:
```
col1 col2 new_col
0 1 4 5
1 2 5 7
2 3 6 9
```
这是一个简单的例子,你可以根据自己的需要编写函数来处理 DataFrame。