g y m 初 步 使 用 gym初步使用 gym初步使用


  • 安装
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple gym

解决Pong-v0环境问题

pip install --no-index -f https://github.com/Kojoley/atari-py/releases atari_py

GYM-OPENAI-环境平台


0.导入工具包
import gym
import random
import numpy as np
import matplotlib.pyplot as plt
1.激活environment , 查看基本信息

1) 激活环境

env = gym.make('Pong-v0')

2) 查看基本信息,包括state信息的shape,action space的大小

# state信息
env.observation_space

gym初步使用_github

# state's shape
env.observation_space.shape

gym初步使用_github_02

# action信息
env.action_space

gym初步使用_环境问题_03

# action个数
env.action_space.n

gym初步使用_初始化_04

2.初始化环境,查看state具体信息
state = env.reset()
state

gym初步使用_github_05

state.shape

gym初步使用_初始化_06

3. 执行action并可视化效果

1) 执行action

state = env.reset()
env.render()

gym初步使用_初始化_07

new_state,reward,done,info = env.step(0)   # 0 表示第一个Action
reward,done,info

gym初步使用_初始化_08

new_state

gym初步使用_初始化_09

new_state -state

gym初步使用_github_10

plt.imshow(state)

gym初步使用_初始化_11

plt.imshow(env.render(mode='rgb_array'))

gym初步使用_github_12

2) 可视化Action

state = env.reset()
done =False
action =3
t = 0
for _ in range(2000):
    t += 1
    env.render()
    state,reward,done,_ = env.step(action)
    if done:
        break
t

gym初步使用_初始化_13

4.关闭环境

env.close()