Matplotlib (pyplot) savefig xuất hình ảnh trống

Tôi đang cố gắng tiết kiệm các lô tôi thực hiện bằng matplotlib; tuy nhiên, hình ảnh đang lưu trống.

Đây là mã của tôi:

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)

Và tessstttyyy.png trống (cũng đã thử với .jpg)

Đầu tiên, điều gì xảy ra khi T0 is not Nonenào? Tôi sẽ kiểm tra điều đó, sau đó tôi sẽ điều chỉnh các giá trị tôi chuyển đến plt.subplot(); có thể thử các giá trị 131, 132 và 133 hoặc các giá trị phụ thuộc vào việc có T0tồn tại hay không .

Thứ hai, sau khi plt.show()được gọi, một con số mới được tạo ra. Để đối phó với điều này, bạn có thể

1. Gọi plt.savefig('tessstttyyy.png', dpi=100)trước khi bạn gọiplt.show()

2. Lưu hình trước khi bạn show()gọi plt.gcf()"lấy số liệu hiện tại", sau đó bạn có thể gọi đối tượng savefig()này Figurebất cứ lúc nào.

Ví dụ:

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

Trong mã của bạn, 'tesssttyyy.png' trống vì nó đang lưu hình mới, không có gì được vẽ.

plt.show() nên đến sau plt.savefig()

Giải thích: plt.show()xóa toàn bộ, vì vậy mọi thứ sau đó sẽ xảy ra trên một hình trống mới

thay đổi thứ tự các chức năng đã khắc phục sự cố cho tôi:

• đầu tiên Lưu cốt truyện
• sau đó hiển thị cốt truyện

như sau:

plt.savefig('heatmap.png')

plt.show()

Gọi savefig trước khi show () làm việc cho tôi.

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

hãy để tôi đưa ra một ví dụ chi tiết hơn:

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