使用decorator保存matplotlib图;保存的输出为空

我正在尝试创建一个修饰函数 @save_fig 围绕 matplotlib 功能( plot_this() )以便自动保存输出。

我可以让装饰器正确执行并显示图形。但是,当 plt.savefig() 将一个空白图形保存到我的目录中。

我想知道我的装饰代码中的逻辑缺少什么?输出应该完全可以从下面的代码中复制。

谢谢你

import matplotlib.pyplot as plt
import pandas as pd

df = pd.DataFrame(data={'foo':list(range(5)), 'bar':list(range(5, 10, 1))})

def save_fig(**param):
    def outer(func):
        def inner(*args, **kwargs):
            ax = func(*args)
            if 'filename' in param.keys():
                print('filename')
                plt.savefig(param['filename'])
            elif 'show' in param.keys():
                print('show')
                plt.show()
            else:
                return ax
        return inner
    return outer

@save_fig(**{'filename': 'foo.png'})
def plot_this():
    plt.scatter(df['foo'], df['bar'])
    plt.show()

if __name__ == "__main__":
    plot_this()