OpenAI gym:安装与简单使用

gym的安装

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

gym的简单例子

(1)

import gym
import time

env = gym.make("CartPole-v1")   # 创建测试环境
env.reset()                     # 设置初始状态
env.render()                    # 显示当前状态的画面

time.sleep(1)                   # 设置当前画面的持续时间

env.close()                     # 关闭测试环境

运行上面的代码,会出现下面的画面,画面保持一段时间后消失
OpenAI gym:安装与简单使用_第1张图片
(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

OpenAI gym:安装与简单使用_第2张图片

参考链接

参考1

你可能感兴趣的:(gym,python,python)