在第一个发现时搜索字符串停止中的关键字

Question

我目前正在尝试在字符串/文本中搜索关键字 +以下单词。我想将这些关键字存储在已经创建的文本文件（f）中。到目前为止，我已经写了此代码：

def keyword_extraction(text, keyword_list, f):

    temp = re.findall(r"[\w']+", text)

    for keyword in keyword_list:
        if keyword in temp:
            results = [temp[temp.index(keyword) + 1]]
            for word in results:
                f.writelines(keyword + ': ' + word + '\n')
        else:
            f.writelines('Keyword "' + keyword + '" not found\n')

问题是，每当找到关键字时，算法就会停止。但是我想提取所有关键字，因此，当它们在文本中两次出现时，应将其写入两次。您对如何解决这个问题有任何建议吗？

示例输入：

text = "today is a sunny day dont you think? I like this day very much"
keyword_list = ['like', 'day']

预期输出：

like: this
day: dont
day: very

实际输出：

like: this
day: dont

谢谢您的帮助！

Answer 1

text = "today is a sunny day dont you think? I like this day very much"
keyword_list = ['like', 'day']

splitted_text = text.split()
for index, word in enumerate(splitted_text):
    if word in keyword_list:
        print(f'{word}: {splitted_text[index+1]}')

输出：

day: dont
like: this
day: very

Answer 2

您可以通过循环和删除从搜索列表中找到的每个关键字的第一次出现，直到不再剩下。

import re

def keyword_extraction(text, keyword_list, f):
    temp = re.findall(r"[\w']+", text)

    for keyword in keyword_list:
        found = False
        while keyword in temp:
            found = True
            try:
                next_word = temp[temp.index(keyword) + 1]
            except IndexError:
                next_word = ''
            f.writelines(keyword + ': ' + next_word + '\n')
            temp.remove(keyword)
        if not found:
            f.writelines('Keyword "' + keyword + '" not found\n')

text = "today is a sunny day dont you think? I like this day very much"
keyword_list = ['like', 'day', 'much']

with open('keyword_search_results.txt', 'w') as f:
    keyword_extraction(text, keyword_list, f)

print('fini')