编写一个函数来执行计算，并将答案作为列添加到现有的数据框中。_编程开发

编写一个函数来执行计算，并将答案作为列添加到现有的数据框中。

创始人

2024-12-06 12:01:04

0次

在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'的新列中。最后，我们打印数据框来查看结果。

上一篇：编写一个函数来指定要发送给ggplot的组/变量组合。

下一篇：编写一个函数来自动化数据提取/返回一个汇总的数据框。

编写一个函数来执行计算，并将答案作为列添加到现有的数据框中。

相关内容

热门资讯