强化学习实战 | 自定义gym环境之显示字符串

如果想用强化学习去实现扫雷、2048这种带有数字提示信息的游戏，自然是希望自定义 gym 环境时能把字符显示出来。上网查了很久，没有找到gym自带的图形工具Viewer可以显示字符串的信息，反而是通过pyglet：

import pyglet
from gym.envs.classic_control import rendering

class DrawText:
    def __init__(self, label:pyglet.text.Label):
        self.label=label
    def render(self):
        self.label.draw()

screen_width = 500
screen_height = 500

viewer = rendering.Viewer(screen_width, screen_height + 20)
text = 'hello world'
label = pyglet.text.Label(text, font_size=36,
                          x=100, y=250, anchor_x='left', anchor_y='bottom',
                          color=(255, 123, 255, 255))
label.draw()
viewer.add_geom(DrawText(label))
viewer.render(return_rgb_array=False)

其中，lable的坐标x y是以左下两边为x y轴（而Viewer是以左上边为轴，这意味着显示图案和文字还得用两套坐标系...），anchor_x 和 anchor_y 指的是label对象的锚点，即如何将label对象视为一个点，有top / bottom / center / baseline四种选择。

效果：

再配合 Python动态变量名定义与调用 的方法，就可以批量地显示字符串了：

import pyglet
from gym.envs.classic_control import rendering

class DrawText:
    def __init__(self, label:pyglet.text.Label):
        self.label=label
    def render(self):
        self.label.draw()

screen_width = 500
screen_height = 500
viewer = rendering.Viewer(screen_width, screen_height)

for i in range(10):
    for j in range(10):
        exec('label_{}_{} = {}'.format(i, j, None))
        names = locals()
        names['label_' + str(i) + '_' + str(j)] = pyglet.text.Label('{}'.format(j), font_size=15,
                                  x=i*50+25, y=j*50+25, anchor_x='left', anchor_y='bottom',
                                  color=(i*10, i*20, i*25, 255))
        label = names['label_{}_{}'.format(i, j)]
        label.draw()
        viewer.add_geom(DrawText(label))

viewer.render(return_rgb_array=False)

效果：