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'))

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