Pandas如何基于DataFrame另一列的值条件替换其他列的匹配值

需求背景 #

现有Pandas DataFrame满足规则：Col1、Col2、Col3三列的行级数值之和，与对应行Main列的数值完全相等。需要实现逻辑：当Col1、Col2、Col3中某单元格取值等于同一行Main列的取值时，将该单元格值替换为字符串EQUAL。

示例原始数据 #

Main        Col1        Col2        Col3
0     100         50          50          0
1     200         0           200         0
2     30          20          5           5
3     500         0           0           500

预期处理结果 #

Main        Col1        Col2        Col3
0     100         50          50          0
1     200         0           EQUAL       0
2     30          20          5           5
3     500         0           0           EQUAL

实现代码 #

推荐使用Pandas内置的向量化方法实现，运行效率高、代码可读性强：

import pandas as pd

# 1. 构造示例原始数据（实际使用时替换为你自己的DataFrame即可）
df = pd.DataFrame({
    'Main': [100, 200, 30, 500],
    'Col1': [50, 0, 20, 0],
    'Col2': [50, 200, 5, 0],
    'Col3': [0, 0, 5, 500]
})

# 2. 定义需要处理的目标列
target_columns = ['Col1', 'Col2', 'Col3']

# 3. 按行匹配条件替换值：目标列值等于同行Main值时，替换为EQUAL
df[target_columns] = df[target_columns].mask(
    df[target_columns].eq(df['Main'], axis=0),
    'EQUAL'
)

如果更习惯逐列处理的写法，也可以用下面的循环实现，三列场景下性能差异可以忽略：

for col in target_columns:
    df.loc[df[col] == df['Main'], col] = 'EQUAL'

注意事项 #

• 替换后Col1/Col2/Col3三列的数据类型会变为object（同时存储数值和字符串），如果后续需要对这三列做数值计算，建议先完成数值相关逻辑再做替换，或者替换后单独提取数值做处理。
• 基于题目前置条件（三列行和等于Main列值），同一行最多只会有一个单元格满足匹配条件，不会出现多列同时替换的冲突问题。

内容的提问来源于stack exchange，提问作者gemlybonid