安装Gym Retro平台

pip install Gym-Retro

# 导入retro模块
import retro
import numpy as np

# 主函数
def main():
    # 创建游戏环境
    env = retro.make(game='Airstriker-Genesis')
    obs = env.reset()  # 初始化游戏环境
    while True:
        # 智能体执行动作，环境返回结果
        obs, rew, done, info = env.step(env.action_space.sample())
        # 刷新当前环境
        env.render()
        if done:
            # 重置游戏环境
            obs = env.reset()
        
        env.close()
    
def reset(self):
    # 利用均匀随机分布初始化环境的状态
    self.state = self.np_random.uniform(low=-0.05, high=0.05, size=(4,))
    # 设置当前步数为None
    self.steps_beyond_done = None
    # 返回环境的初试状态
    return np.array(self.state)
    
# 主函数入口
if __name__ == '__main__':
    main()