Importante

Você está vendo a versão anterior da nova experiência da Alura que estamos preparando para você. Em breve, ela ganha uma identidade visual novinha totalmente pensada em potencializar seus estudos!

0
respostas

import numpy as np

estados = ["Ensolarado", "Nublado", "Chuvoso"]

matriz_transicao = np.array([
[0.7, 0.2, 0.1], # Ensolarado
[0.3, 0.4, 0.3], # Nublado
[0.2, 0.3, 0.5] # Chuvoso
])

def prever_clima(estado_inicial, dias):
estado_atual = estados.index(estado_inicial)
previsao = [estado_inicial]

for _ in range(dias):
    estado_atual = np.random.choice(
        range(len(estados)), 
        p=matriz_transicao[estado_atual]
    )
    previsao.append(estados[estado_atual])

return previsao