MatPlotLib在colorbar之后添加colorbar

Question

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…

我不是很兴奋，因为每个图都有一个额外的颜色条。第一个有自己的own...the，第二个有自己的加上前一个figure....the的，第三个有自己的加上前两个...

我看过这个解决方案，但我真的看不到实现它的好方法。我知道我需要清除之前的颜色栏，如果有的话……我真的真的不明白怎么做。

Answer 1

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