User:Dusan Barok/Markov Chain

import random from pprint import pprint

text = "he saw the black cat before she saw the red potato he hated ." dictionary = {} lastword = ""

for word in text.split():

   if lastword:
       if lastword not in dictionary:
           dictionary[lastword]=[]
       dictionary[lastword].append(word)
   lastword = word

pprint(dictionary)

word = "he" while True:

   print word
   word = random.choice(dictionary[word])
   if word not in dictionary:
       break