pip install gym
推荐使用清华源安装:
pip install gym -i https://pypi.tuna.tsinghua.edu.cn
安装gym相关的依赖包:
pip install gym[all] -i https://pypi.tuna.tsinghua.edu.cn
(1)
import gym
import time
env = gym.make("CartPole-v1") # 创建测试环境
env.reset() # 设置初始状态
env.render() # 显示当前状态的画面
time.sleep(1) # 设置当前画面的持续时间
env.close() # 关闭测试环境
运行上面的代码,会出现下面的画面,画面保持一段时间后消失
(2)
import gym
env = gym.make("CartPole-v1") # 创建环境
for i_episode in range(20):
observation = env.reset() # 保存环境初始状态
for t in range(100):
env.render()
print(observation) # [位置,加速度,角度,角加速度]
action = env.action_space.sample() # 获取一个动作
observation, reward, done, info = env.step(action) # 获取执行上面动作的结果
if done:
print("Episode finished after {} timesteps".format(t + 1))
break
参考1