python - 在 Python 函数中使用 **kwargs

Question

考虑如下方法:

    def find_missing(self, **kwargs):
        if save == True:
            tmp = find_one_missing(self.data)
            tmp.to_csv(path)
        else:
            return find_one_missing(self.data)

我想要实现的是变量save和path为kwargs，用户可以指定其中一个

x.find_missing()

这将返回 find_one_missing() 函数的输出，或者用户也可以插入

x.find_missing(save=True, path=user_string_pathway)

并且输出将自动保存到某个位置。我怎样才能做到这一点？

Answer 1

调用:

find_missing(save=True, path="/path/...")

使用 kwargs['save'] 和 kwargs['path'] 访问值

def find_missing(self, **kwargs):
        if kwargs['save'] == True:
            tmp = find_one_missing(self.data)
            tmp.to_csv(kwargs['path'])
        else:
            return find_one_missing(self.data)