python怎么输入给变量赋值_如何给变量赋值并使用变量作为Python函数的返回值

您可以定义打印功能，如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

然后像这样使用它们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()