python - 在列中搜索单词列表，如果找到单词则创建一个 boolean 列

Question

这个问题已经有答案了:

How to test if a string contains one of the substrings in a list, in pandas? (4 个回答)

已关闭 3 年前。

您好，问题是通过搜索单词列表来循环遍历列，然后如果找到搜索到的任何列表单词，则创建一个 boolean 列。这是我的代码

# Code naf related to sport.
code = ["3230Z","4764Z","7721Z","8551Z","9311Z", "9312Z", "9313Z", "9319Z",
        "9329Z", "364Z" "524W", "714B", "804C", "926A", "926C", "930L", "927C",
        "923K"]

# check keywords of code into "Code_Naf" column
for branch in code:
    df_codeNaf["topNAF"] = df_codeNaf["Code_NAF"].str.contains("3230Z" or "4764Z" or "7721Z" or "8551Z"
                                                                      or "9311Z" or "9312Z" or "9313Z" or "9319Z"
                                                                      or "9329Z" or "364Z" "524W" or "714B" or
                                                                      "804C" or "926A" or "926C" or "930L" or
                                                                      "927C" or "923K")

当我查看 topNaf 列时，我发现只有 2 个 True，但实际上有两个以上。我的代码有什么问题吗？谢谢

Answer 1

您的问题是您对code中的每个banch更改了df_codeNaf['topNAF']。您的代码可以通过以下方式修复:

df_codeNaf['topNAF'] = False
for branch in code:
    df_codeNaf['topNAF'] = df_codeNaf['topNAF'] | df_codeNaf['Code_NAF'].str.contains(branch).

但更好的是，您可以在一行中尝试使用 regex 和 contains :

pattern = '|'.join(code)
df_codeNaf['topNAF'] = df_codeNaf['Code_NAF'].str.contains(pattern)