如何用Python实现：判断单词是否在DataFrame中并生成含Yes的新DataFrame

假设你的DataFrame A包含编号列和存储单词的列，以下是实现需求的代码示例：

import pandas as pd

# 示例DataFrame A
df_a = pd.DataFrame({
    '编号': ['NO001', 'NO002', 'NO003'],
    '单词': ['cat', 'dog', 'bird']
})

x = 'dog'  # 目标单词

# 检查x是否存在于DataFrame的单词列中
if x in df_a['单词'].values:
    # 生成DataFrame B，在匹配行的编号前添加'Yes'
    df_b = df_a.copy()
    df_b.loc[df_b['单词'] == x, '编号'] = 'Yes' + df_b.loc[df_b['单词'] == x, '编号']
else:
    # 若x不存在，可根据需求调整逻辑，此处默认复制原DataFrame
    df_b = df_a.copy()
    print(f"单词{x}不存在于DataFrame A中")

print(df_b)

代码说明 #

• 通过x in df_a['单词'].values快速判断目标单词是否存在
• 匹配成功时，复制原DataFrame得到B，再用loc定位到对应行，对编号列的值做拼接处理
• 若单词不存在，可按需修改分支逻辑（比如抛出提示、返回空DataFrame等）

如果你的DataFrame列名或结构不同，替换代码中对应的列名即可适配。

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