如何检查字符串是否包含列表中的单词

Question

我希望用户输入歌词的程序(这将稍后将扩展到搜索一个网站，但我目前不需要帮助)，该程序将告诉我，如果输入的信息包含一个词从一个列表。

banned_words = ["a","e","i","o","u"] #This will be filled with swear words

profanity = False

lyrics = input ("Paste in the lyrics: ")
for word in lyrics:
    if word in banned_words:
        print("This song says the word "+word)
        profanity = True

if profanity == False:
    print("This song is profanity free")

这段代码只输出“这首歌是不道德的自由”。

Answer 1

有几个想法我会推荐：

• 使用str.split按空格拆分。
• 使用set进行O(1)查找。这是由{}表示的，而不是用于列表的[]。
• 将您的逻辑封装在一个函数中。这样，你就可以简单的return一句脏话就够了。这样您就不再需要else语句了。
• 使用函数意味着不需要设置默认变量，然后在适用的情况下重新分配。
• 使用str.casefold捕获大小写单词。

下面是一个例子：

banned_words = {"a","e","i","o","u"}

lyrics = input("Paste in the lyrics: ")

def checker(lyrics):
    for word in lyrics.casefold().split():
        if word in banned_words:
            print("This song says the word "+word)
            return True
    print("This song is profanity free")
    return False

res = checker(lyrics)