Tạo hàm trả về figure object trong Python Matplotlib

Để có hàm trả về đối tượng figure trong Python (dùng Matplotlib):

• Thiết lập kích thước figure và padding
• Tạo dữ liệu x, y bằng numpy
• Tạo hàm plot(x, y) tạo figure mới bằng figure()
• Vẽ x, y bằng plot(); trả về instance fig
• Gọi plot(x, y) lưu figure vào biến f
• Hiển thị bằng show()

Ví dụ mã nguồn

import numpy as np
from matplotlib import pyplot as plt

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

def plot(x, y):
    fig = plt.figure()
    plt.plot(x, y)
    return fig

x = np.linspace(-10, 10, 100)
y = np.sin(x)
f = plot(x, y)
plt.show()

Kết quả

Đồ thị sin(x) được vẽ từ hàm trả về figure object.

Lợi ích: Cho phép tái sử dụng, chỉnh sửa figure sau khi tạo (lưu file, thêm subplot, chỉnh axes, v.v.) trước khi hiển thị.