FrozenLake4x4-interactive.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import gym

def get_action():
    while True:
        try:
            act = int(input("--- direction(0:L, 1:D, 2:R, 3:U) > "))
            return act
        except ValueError:
            pass

def main():
    env = gym.make("FrozenLake-v0")
    while True:
        ob = env.reset()
        env.render()
        while True:
            ob, reward, done, info = env.step(get_action())
            print()
            env.render()
            print("Observation: {}".format(ob))
            print("Reward: {}".format(reward))
            print("Info: {}".format(info))
            print()
            if done:
                print("========== D O N E ==========")
                print()
                break

if __name__ == "__main__": main()