vis.py 文件源码

def visualize_sensors(state):
    # Clear.
    sns.plt.clf()

    # Make a 2d list.
    cols = [state[0]]

    # Plot it.
    sns.heatmap(data=cols, cmap="Blues_r", yticklabels=False)

    # Draw it.
    sns.plt.draw()

    # Add a pause because you're supposed to.
    sns.plt.pause(0.05)