Analyse Markov

"""
Explicação de como funciona a analise de markov:
https://pastebin.com/nnPSAweC

"""


def analyze_markov(file_name, n=2):
    file = open(file_name)
    dict_of_prefix_suffix = dict()
    list_of_words = list()

    for line in file:
        for word in line.split(" "):
            list_of_words.append(word)

    list_of_prefix = []
    for i in range(n):
        list_of_prefix.append(tuple(list_of_words[i::]))

    list_of_prefix = list(zip(*list_of_prefix))

    position = n

    for words in list_of_prefix:

        if position == len(list_of_words):
            dict_of_prefix_suffix.setdefault(words, []).append(None)
        else:
            dict_of_prefix_suffix.setdefault(words, []).append(list_of_words[position])

        position += 1

    return dict_of_prefix_suffix