Hướng dẫn assign plot to variable python - gán âm mưu cho biến python

Bạn có thể xác định các chức năng âm mưu của mình như

import numpy as np
import matplotlib.pyplot as plt

# an example graph type
def fig_barh(ylabels, xvalues, title=''):
    # create a new figure
    fig = plt.figure()

    # plot to it
    yvalues = 0.1 + np.arange(len(ylabels))
    plt.barh(yvalues, xvalues, figure=fig)
    yvalues += 0.4
    plt.yticks(yvalues, ylabels, figure=fig)
    if title:
        plt.title(title, figure=fig)

    # return it
    return fig

Sau đó sử dụng chúng như

from matplotlib.backends.backend_pdf import PdfPages

def write_pdf(fname, figures):
    doc = PdfPages(fname)
    for fig in figures:
        fig.savefig(doc, format='pdf')
    doc.close()

def main():
    a = fig_barh(['a','b','c'], [1, 2, 3], 'Test #1')
    b = fig_barh(['x','y','z'], [5, 3, 1], 'Test #2')
    write_pdf('test.pdf', [a, b])

if __name__=="__main__":
    main()

Nếu bạn không muốn hình ảnh được hiển thị và chỉ nhận được một biến, thì bạn có thể thử những thứ sau (với một số nội dung bổ sung để xóa trục):

def myplot(t,x):        
    fig = Figure(figsize=(2,1), dpi=80)    
    canvas = FigureCanvasAgg(fig)
    ax = fig.add_subplot()
    ax.fill_between(t,x)
    ax.autoscale(tight=True)
    ax.axis('off')
    canvas.draw()
    buf = canvas.buffer_rgba()
    X = np.asarray(buf)    
    return X

Ví dụ, biến được trả về X có thể được sử dụng với OpenCV và thực hiện

cv2.imshow('',X)

Những nhập khẩu này phải được bao gồm:

from matplotlib.figure import Figure
from matplotlib.backends.backend_agg import FigureCanvasAgg