编写一个通用函数，在pandas中根据另一列的值计算一个特定条件的列值。_编程开发

编写一个通用函数，在pandas中根据另一列的值计算一个特定条件的列值。

创始人

2024-12-07 10:31:16

0次

可以使用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_column(row):
    if row['A'] < 3:
        return row['B'] * 2
    else:
        return row['B'] * 3

# 使用apply方法应用通用函数
df['C'] = df.apply(calculate_column, axis=1)

# 打印结果
print(df)

输出结果：

   A   B   C
0  1   6  12
1  2   7  14
2  3   8  24
3  4   9  27
4  5  10  30

在这个示例中，我们创建了一个包含两列数据的DataFrame。然后，我们定义了一个名为calculate_column的通用函数，该函数根据列A的值计算列B的新值。最后，我们使用apply方法将通用函数应用于DataFrame中的每一行，并将结果保存在新列C中。最终，我们打印整个DataFrame以查看结果。

上一篇：编写一个通用函数，在保存对象时防止覆盖文件。

下一篇：编写一个通用函数：使用复选框禁用一个tkinter小部件。

编写一个通用函数，在pandas中根据另一列的值计算一个特定条件的列值。

相关内容

热门资讯