MatPlotLib在colorbar后添加colorbar

时间:2014-03-25 17:26:33

标签: python matplotlib

remove colorbar from figure in matplotlib

我几乎没有得到如何使用这个答案。

在我的程序中,我需要生成5个图。它们的格式基本相同。这是我的功能:

def plot_that(x_vals, y_vals, z_vals, figname, units, efficiency_or_not):
    ui = uniformity_calc(z_vals)
    if efficiency_or_not:
        plt.scatter(x_vals, y_vals, s = 3*max(x_vals), c = z_vals, cmap = 'rainbow', vmin = 0, vmax = 1)
    else:
        plt.scatter(x_vals, y_vals, s = 3*max(x_vals), c = z_vals, cmap = 'rainbow')
    c = plt.colorbar()
    c.set_label(units)

    plt.xlabel('Uniformity: ' + str(round(ui,2)))
    plt.savefig('./'+ figname + '.jpg', dpi = 100)

我很兴奋它的确有效,最终我可以将所有5个情节都放到我的pdf中...

我不太兴奋每个情节都有1个额外的颜色条。第一个有它自己的...第二个有它自己加上前一个图中的那个....第三个有它自己加上前两个......

我看了这个解决方案,但我并没有真正看到实现它的好方法。我明白我需要清除以前的颜色条,如果有的话...我真的真的不明白怎么做。

1 个答案:

答案 0 :(得分:0)

我添加了一行并解决了我的问题:     plt.close()