Python | Ways to create triplets from given list

Given a list of words, write a Python program to create triplets from the given list.

Examples :

Input: [‘Geeks’, ‘for’, ‘Geeks’, ‘is’, ‘best’, ‘resource’, ‘for’, ‘study’]
Output:
[[‘Geeks’, ‘for’, ‘Geeks’], [‘for’, ‘Geeks’, ‘is’],
[‘Geeks’, ‘is’, ‘best’], [‘is’, ‘best’, ‘resource’],
[‘best’, ‘resource’, ‘for’], [‘resource’, ‘for’, ‘study’]]

Input: [‘I’, ‘am’, ‘Paras’, ‘Jain’, ‘I’, ‘Study’, ‘From’, ‘GFG’]
Output:
[[‘I’, ‘am’, ‘Paras’], [‘am’, ‘Paras’, ‘Jain’],
[‘Paras’, ‘Jain’, ‘I’], [‘Jain’, ‘I’, ‘Study’],
[‘I’, ‘Study’, ‘From’], [‘Study’, ‘From’, ‘GFG’]]

Let’s see some of the methods to do this task.

Method #1: Using List comprehension

# Python code to create triplets from list of words.

# List of word initialization
list_of_words = ['I', 'am', 'Paras', 'Jain',
                 'I', 'Study', 'DS', 'Algo']

# Using list comprehension
List = [list_of_words[i:i + 3]
        for i in range(len(list_of_words) - 2)]

# printing list
print(List)

Output:

[['I', 'am', 'Paras'], ['am', 'Paras', 'Jain'], 
 ['Paras', 'Jain', 'I'], ['Jain', 'I', 'Study'],
 ['I', 'Study', 'DS'], ['Study', 'DS', 'Algo']]

Method #2: Using Iteration

# Python code to create triplets from list of words.

# List of word initialization
list_of_words = ['Geeks', 'for', 'Geeks', 'is',
                 'best', 'resource', 'for', 'study']

# Output list initialization
out = []

# Finding length of list
length = len(list_of_words)

# Using iteration
for z in range(0, length-2):
    # Creating a temp list to add 3 words
    temp = []
    temp.append(list_of_words[z])
    temp.append(list_of_words[z + 1])
    temp.append(list_of_words[z + 2])
    out.append(temp)

# printing output
print(out)

Output:

[['Geeks', 'for', 'Geeks'], ['for', 'Geeks', 'is'],
 ['Geeks', 'is', 'best'], ['is', 'best', 'resource'],
 ['best', 'resource', 'for'], ['resource', 'for', 'study']]