python - 我可以在 python 3.x 的函数内部动态切换函数吗？

Question

在 Python 3.x 中，是否可以在函数或类中动态选择函数？例如在下面的函数中，可以不依赖条件语句而只输入我想使用的方法吗？

def npMethodChooser(n,m,method):
    if method == "mean":
        return np.random.rand(n,m).mean(axis=1)
    if method == "std":
        return np.random.rand(n,m).std(axis=1)

Answer 1

可以使用getattr 函数进行动态属性查找。

def npMethodChooser(n, m, method):
    return getattr(np.random.rand(n,m), method)(axis=1)