Pandas布尔掩码创建及字符串替换与行标记问题求助

问题说明 #

我是Python新手，要处理一个DataFrame：对A、B、E列中以"dog"或"cat"结尾的单元格替换为空字符串，并将对应行的'Flag'列设为1。但自己写的掩码代码无法正常工作。

原DataFrame结构 #

import pandas as pd

df = pd.DataFrame({
    'A': ['A-dog', "B-dog","C-cat" , "E-snake", "F-hamser"],
    'B': ['F-dog', "B-parrot","C-snake" , "E-cat", "F-bird"],
    'C': [1, 2, 3, 4, 5],
    'D': [22,23,24,25,26],
    'E': ['A-snake', "B-dog","C-snake" , "E-snake", "F-snake"],
    'Flag': [0,0,0,0,0]
})

尝试的错误代码 #

cols=['A','B','E']
mask=df[cols].apply(lambda x: 'dog'  or 'cat' in x[-3:])
# x[-3:] to select the last three characters of the string.
# If the mask were working, I would change the flag variable in this way
df.loc(mask.any(axis=1),'Flag')=1

期望结果 #

res = pd.DataFrame({
    'A': ['A-', "B-","C-" , "E-snake", "F-hamser"],
    'B': ['F-', "B-parrot","C-snake" , "E-", "F-bird"],
    'C': [1, 2, 3, 4, 5],
    'D': [22,23,24,25,26],
    'E': ['A-snake', "B-","C-snake" , "E-snake", "F-snake"],
    'Flag': [1,1,1,1,0]
})

问题排查 #

你的代码有两个关键错误：

1. Lambda逻辑错误：'dog' or 'cat' in x[-3:] 中，'dog'是是非空字符串，布尔值永远为True，导致掩码全为True，无法正确匹配目标字符串。正确逻辑应为判断x[-3:]是否包含'dog'或'cat'，即'dog' in x[-3:] or 'cat' in x[-3:]。
2. loc语法错误：df.loc()应该使用方括号df.loc[]，圆括号会导致语法错误。

正确解决方案 #

import pandas as pd

# 初始化DataFrame
df = pd.DataFrame({
    'A': ['A-dog', "B-dog","C-cat" , "E-snake", "F-hamser"],
    'B': ['F-dog', "B-parrot","C-snake" , "E-cat", "F-bird"],
    'C': [1, 2, 3, 4, 5],
    'D': [22,23,24,25,26],
    'E': ['A-snake', "B-dog","C-snake" , "E-snake", "F-snake"],
    'Flag': [0,0,0,0,0]
})

cols = ['A', 'B', 'E']

# 1. 创建正确的布尔掩码：检查单元格是否以dog或cat结尾
mask = df[cols].apply(lambda x: x.str[-3:].isin(['dog', 'cat']))

# 2. 替换符合条件的单元格为空字符串（保留前半部分）
df[cols] = df[cols].mask(mask, df[cols].str[:-3])

# 3. 将存在匹配项的行Flag设为1
df.loc[mask.any(axis=1), 'Flag'] = 1

print(df)

运行后输出与期望结果一致：

A          B  C   D          E  Flag
0        A-        F-  1  22  A-snake     1
1        B-  B-parrot  2  23        B-     1
2        C-  C-snake  3  24  C-snake     1
3  E-snake        E-  4  25  E-snake     1
4  F-hamser    F-bird  5  26  F-snake     0

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