可视化 Mountain Car 环境

现在，您将进一步探索 Mountain Car 环境。可视化是理解强化学习环境动态的关键一步。您将编写一个 render() 函数，用于显示环境的当前状态。之后，您可以在任何想要可视化的环境中复用这个函数。

matplotlib.pyplot 和 gymnasium 已分别以 plt 和 gym 导入。

本练习是课程的一部分

Python 中的 Gymnasium 强化学习

查看课程

练习说明

完成 render() 函数以可视化环境，获取环境的 state_image 并将其绘制出来。
调用 render() 函数以显示环境的当前状态。

交互式实操练习

通过完成这段示例代码来试试这个练习。

env = gym.make('MountainCar', render_mode='rgb_array')
initial_state, _ = env.reset()

# Complete the render function
def render():
    state_image = ____
    ____
    plt.show()

# Call the render function    
____

编辑并运行代码