MatPlotLib añadiendo barra de colores después de la barra de colores

eliminar barra de colores de la figura en matplotlib

No estoy entendiendo cómo usar esta respuesta en la práctica.

En mi progtwig, necesito producir 5 plots. Todos ellos son básicamente el mismo formato. Aquí está mi función:

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) 

Estoy muy emocionado de que funcione y, en última instancia, puedo incluir los 5 gráficos en mi pdf …

Estoy menos emocionado de que cada plot tenga 1 barra de colores adicional. El primero tiene su propio … el segundo tiene su propio más el de la figura anterior … el tercero tiene su propio más los dos anteriores …

Miré esta solución, pero realmente no veo una buena manera de implementarla. Entiendo que necesito borrar la barra de colores anterior si hay una … Realmente no entiendo cómo.

Agregué una línea y resolví mi problema: plt.close ()