plot.py 文件源码

def action_distribution(actions, ax=None, file="action_ditribution.png"):
    plt.figure(figsize=(10, 10))
    sb.distplot(actions, kde=False, ax=ax)
    plt.ylabel("probability")
    plt.xlabel("action")
    plt.title("Action distribution")
    plt.savefig(file)
    plt.close()