python - Matplotlib 工作流程

Question

我正在做一个学校项目，我正在使用 matplotlib 绘制我所有的图，但我发现复制第一个图然后更改变量名称以添加其他图(子图)非常乏味。

你可以在这个链接上看到我是如何做到的:

IPython Project Notebook

非常感谢有关如何改进我的工作流程的反馈，谢谢。

Answer 1

考虑将绘图机制隔离到它们自己的函数中，这样您就可以创建一个 for 循环，为每个变量调用该函数。这是一个示例(我只复制了您的两个参数列表，但您可以看到如何添加其余参数)。

fig, ax = plt.subplots(nrows=3, ncols= 2, figsize=(10,10))
x = np.linspace(0,1000000,num=10000)

def plot_maker(ax, x, var, var_name, ax_x, ax_y):
    sh, l, sc = sts.weibull_min.fit(var, floc=0)
    y = sts.weibull_min.pdf(x, sh, l, sc)
    print('Weibull parameters for {0}:\n Scale: {1:2.2f}\n Shape: {2:02.2f}'.format(var_name, sc, sh))
    ax[ax_x, ax_y].hist(var, normed=True)
    ax[ax_x, ax_y].plot(x, y, 'r', lw=2)

params = [
    [fuel['TTF or TBF'], 'Fuel TBF', 0, 0],
    [engine['TBF(TTF)'], 'Engine TBF', 0, 1],
]

for param_list in params:
    var, var_name, ax_x, ax_y = param_list
    plot_maker(ax, x, var, var_name, ax_x, ax_y)

有很多种写法，但重要的是将建模和绘图与各个参数分开，因此您可以根据需要添加任意参数。