python - Python：将函数作为参数传递，并设置选项

Question

在 Python 中，我需要在相同的输入参数 sampleA上调用许多非常相似的函数，并且sampleB. 唯一的问题是，其中一些功能需要设置一个选项，而有些则不需要。例如：

import scipy.stats
scipy.stats.mannwhitneyu(sampleA, sampleB)
[...some actions...]
scipy.stats.mstats.ks_twosamp(sampleA, sampleB, alternative='greater')
[...same actions as above...]
scipy.stats.mstats.mannwhitneyu(sampleA, sampleB, use_continuity=True)
[...same actions as above...]

因此，我想将此类函数的名称作为更通用函数的输入参数computeStats以及sampleAand传递sampleB，但我不知道如何处理有时必须使用的选项。

def computeStats(functionName, sampleA, sampleB, options???):
   functionName(sampleA, sampleB)  #and options set when necessary
   ...some actions...
   return testStatistic

如何指定有时必须设置、有时不需要设置的选项？

Answer 1

使用**kwargs：

def computeStats(func, sampleA, sampleB, **kwargs):
   func(sampleA, sampleB, **kwargs)
   ...some actions...
   return testStatistic

然后你就可以computeStats()像这样使用：

computeStats(scipy.stats.mstats.ks_twosamp, sampleA, sampleB, alternative='greater')

也就是说，我并不完全相信你需要这个。简单点怎么样

def postprocessStats(testStatistic):
   ...some actions...
   return testStatistic

postprocessStats(scipy.stats.mstats.ks_twosamp(sampleA, sampleB, alternative='greater'))

?

我认为这更容易阅读，同时也更通用。