Matplotlib (pyplot) savefig genera una imagen en blanco

Estoy tratando de guardar tramas que hago usando matplotlib; sin embargo, las imágenes se guardan en blanco.

Aquí está mi código:

plt.subplot(121)
plt.imshow(dataStack, cmap=mpl.cm.bone)

plt.subplot(122)
y = copy.deepcopy(tumorStack)
y = np.ma.masked_where(y == 0, y)

plt.imshow(dataStack, cmap=mpl.cm.bone)
plt.imshow(y, cmap=mpl.cm.jet_r, interpolation='nearest')

if T0 is not None:
    plt.subplot(123)
    plt.imshow(T0, cmap=mpl.cm.bone)

    #plt.subplot(124)
    #Autozoom

#else:
    #plt.subplot(124)
    #Autozoom

plt.show()
plt.draw()
plt.savefig('tessstttyyy.png', dpi=100)

Y tessstttyyy.png está en blanco (también probado con .jpg)

Primero, ¿qué pasa cuando T0 is not None? Lo probaría, luego ajustaría los valores a los que paso plt.subplot(); quizás pruebe los valores 131, 132 y 133, o valores que dependen de si T0existen o no .

En segundo lugar, después de que plt.show()se llama, se crea una nueva figura. Para lidiar con esto, puedes

1. Llama plt.savefig('tessstttyyy.png', dpi=100)antes de llamarplt.show()

2. Guarde la cifra que tiene delante show()llamando plt.gcf()a "obtener la cifra actual", luego puede llamar savefig()a este Figureobjeto en cualquier momento.

Por ejemplo:

fig1 = plt.gcf()
plt.show()
plt.draw()
fig1.savefig('tessstttyyy.png', dpi=100)

En su código, 'tesssttyyy.png' está en blanco porque está guardando la nueva figura, a la que no se ha trazado nada.

plt.show() debería venir después plt.savefig()

Explicación: plt.show()borra todo el asunto, por lo que cualquier cosa posterior sucederá en una nueva figura vacía

cambiar el orden de las funciones me solucionó el problema :

• primero guardar la trama
• luego muestra la trama

de la siguiente manera:

plt.savefig('heatmap.png')

plt.show()

Llamar a savefig antes de show () funcionó para mí.

fig ,ax = plt.subplots(figsize = (4,4))
sns.barplot(x='sex', y='tip', color='g', ax=ax,data=tips)
sns.barplot(x='sex', y='tip', color='b', ax=ax,data=tips)
ax.legend(['Male','Female'], facecolor='w')

plt.savefig('figure.png')
plt.show()

déjenme dar un ejemplo más detallado:

import numpy as np
import matplotlib.pyplot as plt


def draw_result(lst_iter, lst_loss, lst_acc, title):
    plt.plot(lst_iter, lst_loss, '-b', label='loss')
    plt.plot(lst_iter, lst_acc, '-r', label='accuracy')

    plt.xlabel("n iteration")
    plt.legend(loc='upper left')
    plt.title(title)
    plt.savefig(title+".png")  # should before plt.show method

    plt.show()


def test_draw():
    lst_iter = range(100)
    lst_loss = [0.01 * i + 0.01 * i ** 2 for i in xrange(100)]
    # lst_loss = np.random.randn(1, 100).reshape((100, ))
    lst_acc = [0.01 * i - 0.01 * i ** 2 for i in xrange(100)]
    # lst_acc = np.random.randn(1, 100).reshape((100, ))
    draw_result(lst_iter, lst_loss, lst_acc, "sgd_method")


if __name__ == '__main__':
    test_draw()