Python在窗口大小条件下提取包含2个单词的句子

Question

我遇到了与此链接中讨论的相同的问题 Python extracting sentence containing 2 words

但不同之处在于我只需要在定义大小的窗口搜索中提取包含这两个单词的句子。例如：

sentences = [ 'There was peace and happiness','hello every one',' How to Find Inner Peace ,love and Happiness ','Inner peace is closely related to happiness']

search_words= ['peace','happiness']
windows_size = 3  #search only the three words after the 1est word 'peace'
#output must be :
output= ['There was peace and happiness',' How to Find Inner Peace love and Happiness ']

Answer 1

这是一个粗略的解决方案。

def search(sentences, keyword1, keyword2, window=3):
    res = []
    for sentence in sentences:
        words = sentence.lower().split(" ")
        if keyword1 in words and keyword2 in words:
            keyword1_idx = words.index(keyword1)
            keyword2_idx = words.index(keyword2)
            if keyword2_idx - keyword1_idx <= window:
                res.append(sentence)
    return res

给定一个sentences 列表和两个关键字keyword1 和keyword2，我们一个一个地遍历sentences 列表。我们将句子拆分为单词，假设单词由一个空格分隔。然后，在对words 列表中是否存在两个关键字进行粗略检查后，我们在words 中找到每个关键字的索引，以确保索引最多相隔window，即 单词在window 单词中靠得很近。我们只将满足此条件的句子附加到res 列表中，并返回该结果。