Python | Check if given words appear together in a list of sentence

Given a list of sentences ‘sentence’ and a list of words ‘words’, write a Python program to find which sentence in the list of sentences consist of all words contained in ‘words’ and return them within a list. 

Examples:

Input : sentence = ['I love tea', 'He hates tea', 'We love tea']
        words = ['love', 'tea']
Output : ['I love tea', 'We love tea']

Input : sentence = ['python coder', 'geeksforgeeks', 'coder in geeksforgeeks']
        words = ['coder', 'geeksforgeeks']
Output : ['coder in geeksforgeeks']

Approach #1 : Using List comprehension We first use list comprehension, to return a boolean value for each substring of the list of sentence and store it in ‘res’. Finally, return a list comprising of the desired sentences according to the boolean values in ‘res’. 

Python3

# Python3 program to Check if given words
# appear together in a list of sentence


def check(sentence, words):
    res = [all([k in s for k in words]) for s in sentence]
    return [sentence[i] for i in range(0, len(res)) if res[i]]


# Driver code
sentence = ['python coder', 'geeksforgeeks', 'coder in geeksforgeeks']
words = ['coder', 'geeksforgeeks']
print(check(sentence, words))

Output

['coder in geeksforgeeks']

Approach #2 : List comprehension (Alternative way) For each substring in list of sentences, it checks how many words are there in the current substring and stores it in a variable ‘k’. If the length of ‘k’ matches with length of list of words, just append it to ‘res’. 

Python3

# Python3 program to Check if given words
# appear together in a list of sentence


def check(sentence, words):
    res = []
    for substring in sentence:
        k = [w for w in words if w in substring]
        if (len(k) == len(words)):
            res.append(substring)

    return res


# Driver code
sentence = ['python coder', 'geeksforgeeks', 'coder in geeksforgeeks']
words = ['coder', 'geeksforgeeks']
print(check(sentence, words))

Output

['coder in geeksforgeeks']

Approach #3 : Python map() map() method applies a function on list of sentences and check if all words are contained in the list or not by splitting the list of words. It returns a boolean value for each substring of the list of sentence and store it in ‘res’. Finally, repeat the same steps as in approach #1. 

Python3

# Python3 program to Check if given words
# appear together in a list of sentence


def check(sentence, words):
    res = list(map(lambda x: all(map(lambda y: y in x.split(),
                                     words)), sentence))
    return [sentence[i] for i in range(0, len(res)) if res[i]]


# Driver code
sentence = ['python coder', 'geeksforgeeks', 'coder in geeksforgeeks']
words = ['coder', 'geeksforgeeks']
print(check(sentence, words))

Output

['coder in geeksforgeeks']

Approach #4 : Using count()+len()+for loop

Python3

# Python3 program to Check if given words
# appear together in a list of sentence


def check(sentence, words):
    res = []
    for i in sentence:
        c = 0
        for j in words:
            if(i.count(j) >= 1):
                c += 1
        if(c == len(words)):
            res.append(i)
    return res


# Driver code
sentence = ['python coder', 'geeksforgeeks', 'coder in geeksforgeeks']
words = ['coder', 'geeksforgeeks']
print(check(sentence, words))

Output

['coder in geeksforgeeks']

Approach #4 : Using built-in issubset

This solution converts both the list of sentences and the list of words into sets and checks if the list of words is a subset of any of the sentence sets. If it is, it adds the sentence to the result list. This approach is faster than the previous ones because it uses the built-in issubset function which has a time complexity of O(n).

Python3

def check(sentence, words):
    # list to store sentences that contain all words
    res = []
    # convert list of words into set for faster lookup
    words_set = set(words)
    # iterate through each sentence
    for s in sentence:
        # convert sentence into set for faster lookup
        sentence_set = set(s.split())
        # check if list of words is a subset of sentence set
        # if it is, add sentence to result list
        if words_set.issubset(sentence_set):
            res.append(s)
    return res

# Driver code
sentence = ['python coder', 'geeksforgeeks', 'coder in geeksforgeeks']
words = ['coder', 'geeksforgeeks']
print(check(sentence, words))
#This code is contributed by Edula Vinay Kumar Reddy

Output

['coder in geeksforgeeks']

Time complexity: O(n*m) where n is the number of sentences and m is the length of the longest sentence
Auxiliary Space: O(n*m)