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