Làm thế nào để chuyển đổi Matplotlib figure sang đối tượng PIL Image?

Để chuyển đổi hình matplotlib thành đối tượng hình ảnh PIL, chúng ta có thể thực hiện các bước sau -

• Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô phụ.
• Tạo một hình mới hoặc kích hoạt một hình hiện có.
• Vẽ một danh sách bằng cách sử dụng plot () phương pháp.
• Khởi tạo bộ đệm trong bộ nhớ.
• Lưu hình ảnh trong bộ đệm.
• Sử dụng hình ảnh PIL để lấy đối tượng hình ảnh.
• Hiển thị hình ảnh hiện tại.
• Đóng bộ đệm I / O trong bộ nhớ.

Ví dụ

import io
from PIL import Image
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

plt.figure()
plt.plot([1, 2])

img_buf = io.BytesIO()
plt.savefig(img_buf, format='png')

im = Image.open(img_buf)
im.show(title="My Image")

img_buf.close()

Đầu ra