Matplotlib (pyplot) savefig menampilkan gambar kosong

Saya mencoba menyimpan plot yang saya buat menggunakan matplotlib; Namun, gambar-gambarnya kosong.

Ini kode saya:

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)

Dan tessstttyyy.png kosong (juga dicoba dengan .jpg)

Pertama, apa yang terjadi kapan T0 is not None? Saya akan menguji itu, kemudian saya akan menyesuaikan nilai-nilai yang saya berikan plt.subplot(); mungkin mencoba nilai 131, 132, dan 133, atau nilai yang bergantung pada ada atau tidaknya T0.

Kedua, setelah plt.show()dipanggil, sosok baru dibuat. Untuk mengatasinya, Anda bisa

1. Panggil plt.savefig('tessstttyyy.png', dpi=100)sebelum meneleponplt.show()

2. Simpan gambar sebelum Anda show()memanggil plt.gcf()"dapatkan gambar saat ini", maka Anda dapat memanggil objek savefig()ini Figurekapan saja.

Sebagai contoh:

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

Dalam kode Anda, 'tesssttyyy.png' kosong karena menyimpan angka baru, yang tidak ada yang diplot.

plt.show() harus datang setelah plt.savefig()

Penjelasan: plt.show()menghapus semuanya, jadi apa pun setelahnya akan terjadi pada sosok kosong baru

ubah urutan fungsi yang memperbaiki masalah untuk saya:

• pertama Simpan plot
• lalu Perlihatkan plotnya

sebagai berikut:

plt.savefig('heatmap.png')

plt.show()

Memanggil savefig sebelum show () bekerja untuk saya.

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()

izinkan saya memberi contoh yang lebih detail:

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()