在同一个文件 matplotlib 中绘制和保存多个函数

Question

我想使用 matplotlib 在一页中保存 6 个图形。这些图表都是对一个函数的调用，并在保存之前提出了以下代码进行测试：

def save_plot (output_DF, path = None):
    fig = plt.figure(1)
    sub1 = fig.add_subplot(321)
    plt.plot(plot_BA(output_DF))
    sub2 = fig.add_subplot(322)
    sub2.plot(plot_merchantableVol(output_DF))
    sub3 = fig.add_subplot(323)
    sub3.plot(plot_topHeight(output_DF))
    sub4 = fig.add_subplot(324)
    sub4.plot(plot_GrTotVol(output_DF))
    sub5 = fig.add_subplot(325)
    sub5.plot(plot_SC(output_DF))
    sub6 = fig.add_subplot(326)
    sub6.plot(plot_N(output_DF))
    plt.show()

实际上，我确实创建了一个包含 6 个空图的页面，但也为我调用的每个函数创建了 6 个单独的图。例如，plot_BA(output_DF) 是一个函数，我调用它来读取一个 csv 文件并创建一个绘图（单独它正在工作）。另一个是类似的功能，并且也可以正常工作。似乎我错过了将图表放在他们指定的无花果位置的东西。

这是我正在使用的功能之一。

def  plot_BA(output_DF):
    BA =  output_DF.loc[:,['BA_Aw','BA_Sw', 'BA_Sb','BA_Pl']]    
    BAPlot = BA.plot()
    plt.xlabel('Year', fontsize=14)
    plt.ylabel('BA (m2)')
    return True

有什么建议吗？

Answer 1

试试这样的：

import matplotlib.pyplot as plt
import random

def function_to_call(func_name, ax):
    data = range(10)
    random_x_label = random.choice('abcdefghijk')
    random_y_label = random.choice('abcdefghijk')
    random.shuffle(data)  # demonstration

    ax.plot(data)
    ax.set_xlabel(random_x_label)
    ax.set_ylabel(random_y_label)
    ax.set_title(func_name)


fig = plt.figure(1)
sub1 = fig.add_subplot(321)
function_to_call('call_0', sub1)
sub2 = fig.add_subplot(322)
function_to_call('call_1', sub2)
sub3 = fig.add_subplot(323)
function_to_call('call_2', sub3)
sub4 = fig.add_subplot(324)
function_to_call('call_3', sub4)
sub5 = fig.add_subplot(325)
function_to_call('call_4', sub5)
sub6 = fig.add_subplot(326)
function_to_call('call_5', sub6)
plt.tight_layout()  # just to improve spacings
plt.show()
fig.savefig('output_plot.png')

这个想法是将您的轴对象（这是整个图中的众多对象之一）转发给您的函数。然后，您将需要轴级功能。注意plt.xlabel（figure-level）和ax.set_xlabel（axes-level）之间的区别。