Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách chuyển đổi figure Matplotlib thành đối tượng PIL Image?


Để chuyển đổi figure Matplotlib thành đối tượng PIL Image, chúng ta có thể thực hiện các bước sau:

  • Thiết lập kích thước figure và điều chỉnh padding giữa và xung quanh subplots.
  • Tạo figure mới hoặc kích hoạt figure hiện có.
  • Vẽ danh sách bằng phương thức plot().
  • Khởi tạo buffer trong bộ nhớ.
  • Lưu hình ảnh buffer.
  • Dùng PIL Image để lấy đối tượng hình ảnh.
  • Hiển thị hình ảnh hiện tại.
  • Đóng buffer 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()

Kết quả

Cách chuyển đổi figure Matplotlib thành đối tượng PIL Image?Cách chuyển đổi figure Matplotlib thành đối tượng PIL Image?