def get2GramSentence(word, n = 50): for i in range(n): print (word), # Find Next word word = next((element[0][1] for element in gram2 if element[0][0] == word),None ) if not word: break print(word) word = start_word get2GramSentence(word, 20)