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 以及 sampleA 和 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'))

?

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