23 lines
573 B
Python
23 lines
573 B
Python
import gymnasium as gym
|
|
import time
|
|
|
|
# 创建环境
|
|
env = gym.make("CartPole-v1", render_mode="human") # human模式会用pyglet显示窗口
|
|
|
|
# 重置环境
|
|
observation, info = env.reset()
|
|
print("初始观察值:", observation)
|
|
print("<UNK>:", info)
|
|
|
|
# 随机动作走几步
|
|
for step in range(100):
|
|
observation, reward, terminated, truncated, info = env.step(1)
|
|
print(f"Step {step+1}: observation={observation}, reward={reward}, done={terminated}")
|
|
|
|
if terminated or truncated:
|
|
observation, info = env.reset()
|
|
|
|
time.sleep(1)
|
|
# 关闭环境
|
|
env.close()
|