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
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
# state's shape
env.observation_space.shape
# action信息
env.action_space
# action个数
env.action_space.n
2.初始化环境,查看state具体信息
state = env.reset()
state
state.shape
3. 执行action并可视化效果
1) 执行action
state = env.reset()
env.render()
new_state,reward,done,info = env.step(0) # 0 表示第一个Action
reward,done,info
new_state
new_state -state
plt.imshow(state)
plt.imshow(env.render(mode='rgb_array'))
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
4.关闭环境
env.close()