Python：在给定两个预先生成的函数的情况下，如何有效地选择要使用的函数？

Question

假设我有一个带有两个函数的模块myfunctions：

def functionA():
    return'Function A chosen.'

def functionB():
    return 'Function B chosen.'

然后，我有一个模块silly，它对以下函数执行一些操作：

def sillyprinter():
    print chosen_function
    print chosen_function()

然后，在我的主要剧本中，我有：

import myfunctions
import silly

class FunctionChooser():
    def __init__(function_choice="A"):
        self.choose_function_given_choice(function_choice)

    def do_something_many_times(self):
        for i in range(1000):
            silly.sillyprinter()

FunctionChooser的思想是用一些用户给定的信息来初始化它，这些信息与我想从myfunctions使用哪个函数（A或B）有关。然后，在做出这个选择之后，用户可以调用do_something_many_times，这将使用另一个模块，该模块将多次使用从myfunctions中选择的函数。你知道吗

为了提高效率，我想避免以下情况：

1）一次又一次地选择要使用的函数——所以我想做一次选择（在初始化期间），然后以某种方式“保存”它。此要求取消了以下设计的资格：

def sillyprinter(chosen_function):
    print chosen_function
    print chosen_function()

# =================================

class FunctionChooser():
    def __init__(function_choice="A"):
        self.function_choice = function_choice

    def choose_function_given_choice(self):
        if self.function_choice == "A":
            return myfunctions.functionA
        elif self.function_choice == "B":
            return myfunctions.functionB

    def do_something_many_times(self):
        for i in range(1000):
            silly.silly_printer(self.choose_function_given_choice())

2）存储我想选择作为类属性的函数since ^{} calls are expensive, and the use case is numerical。这一要求取消了如下设计的资格：

def sillyprinter(chosen_function):
    print chosen_function
    print chosen_function()

# =================================

class FunctionChooser():
    def __init__(function_choice="A"):
        self.function_choice = function_choice

    def choose_function_given_choice(self):
        if self.function_choice == "A":
            self.chosen_function = myfunctions.functionA
        elif self.function_choice == "B":
            self.chosen_function = myfunctions.functionB

    def do_something_many_times(self):
        for i in range(1000):
            silly.silly_printer(self.chosen_function)

---

我目前的想法是：

def sillyprinter(chosen_function):
    print chosen_function
    print chosen_function()

# =================================

class FunctionChooser():
    def __init__(function_choice="A"):
        self.function_choice = function_choice

    def choose_function_given_choice(self):
        if self.function_choice == "A":
            self.chosen_function = myfunctions.functionA
        elif self.function_choice == "B":
            self.chosen_function = myfunctions.functionB

    def do_something_many_times(self):
        chosen_function = self.chosen_function
        for i in range(1000):
            silly.silly_printer(chosen_function)

它只是与我在2）中取消的设计稍有不同，因为它只进行一次self调用，然后将该函数存储为局部变量。你知道吗

Is this the best way to design such "function choice" in Python, given my requirements? (see below)

编辑：我想我应该把要求弄清楚，这样问题就不那么笼统了

1. “函数选择字符串”不必是字符串——它可以是为某种函数选择编码的任何变量。你知道吗
2. 我想避免self.访问调用（但通常可能是.访问调用？）由于numba使用访问调用优化代码的问题：Design heuristics for writing Python classes that interact with `scipy.integrate.odeint`?
3. “function choice”变量只给出一次，因此我们可以认为它实际上是一个常量。你知道吗

Answer 1

不，Python中的函数是一级对象；将它们视为一级对象。你知道吗

class FunctionChooser():
    def __init__(function_choice=myfunctions.functionA):
        self.function_choice = function_choice

    def do_something_many_times(self):
        chosen_function = self.chosen_function
        printer = silly.silly_printer
        for i in range(1000):
            printer(chosen_function)