在Python中,可以使用Pandas库来操作数据框,并使用apply函数将函数应用于数据框的每一行。以下是一个示例代码,演示如何编写一个函数来执行计算,并将答案作为列添加到现有的数据框中:
import pandas as pd
# 创建示例数据框
data = {'A': [1, 2, 3, 4, 5],
'B': [6, 7, 8, 9, 10]}
df = pd.DataFrame(data)
# 定义计算函数
def calculate(row):
# 在此处执行计算
answer = row['A'] + row['B']
return answer
# 将函数应用于数据框的每一行,并将答案添加到新列中
df['Answer'] = df.apply(calculate, axis=1)
# 打印结果
print(df)
输出结果如下:
A B Answer
0 1 6 7
1 2 7 9
2 3 8 11
3 4 9 13
4 5 10 15
在这个示例中,我们定义了一个名为calculate的函数,该函数接受一行数据作为输入,并执行计算。在这种情况下,计算是将'A'列和'B'列的值相加。然后,我们使用apply函数将calculate函数应用于数据框的每一行,并将计算结果添加到名为'Answer'的新列中。最后,我们打印数据框来查看结果。