创建动态类方法的 Pythonic 方式

Question

我想知道以下策略是否是在方法中创建动态函数的正确/pythonic 方式。目标是拥有一个可以根据FUN() 定义的复杂模型计算值的类，但我希望能够在脚本中更改该模型而不重写它，或者创建一堆不同类型的类（因为这个函数是我唯一希望改变的）。

我还在回复question 时读到，我的结构方式可能最终会变慢？我打算调用setupFunction() 1 到3 次模拟（更改模型）并调用FUN 数千次。

伪代码...

class MyClass:

  def __init__(self, model = 'A'):
    self.setupFunction(model)
    # Other Class Stuff...

  def setupFunction(self, _model):
    if _model == 'A':
      def myFunc(x):
        # Do something with x, store in result
        return result
    else:
      def myFunc(x):
        # Do something different with x
        return result
    self.FUN = myFunc

  # Other Class Methods... some of which may call upon self.FUN


Model1 = MyClass('A')
Model2 = MyClass('B')
print(Model1.FUN(10))
print(Model2.FUN(10))

我做了一些小测试，上面的内容乍一看似乎并没有中断。我知道我也可以通过执行以下操作来做类似的事情，但是它必须在每次调用 FUN() 时测试模型，最后我会有许多不同的模型案例：

class MyClass():

  def __init__(self, model = 'A'):

  def FUN(self, x):
    if self.model == 'A':
      # result = Stuff
    else:
      # result = Other Stuff
    return result

对 python 还是很陌生，所以感谢任何反馈！

Answer 1

不确定我是否理解您的问题... 像这样的东西呢？

class MyClass():

    model_func = {'A' : funca, 'B' : funcb}

    def __init__(self, model):
        self.func = self.model_func[model]

    def funca():
        pass

    def funcb():
        pass

a = MyClass('A')
a.func()

b = MyClass('B')
b.func()

其他选项可能是这样的（更好的关注点分离）：

class Base(object):

    def __new__(cls, category, *arguments, **keywords):
        for subclass in Base.__subclasses__():
            if subclass.category == category:
                return super(cls, subclass).__new__(subclass, *arguments, **keywords)
        raise Exception, 'Category not supported!'

class ChildA(Base):
    category = 'A'

    def __init__(self, *arguments, **keywords):
        print 'Init for Category A', arguments, keywords

    def func(self):
        print 'func for Category A'


class ChildB(Base):
    category = 'B'

    def func(self):
        print 'func for Category B'

if __name__ == '__main__':
    a = Base('A')
    a.func()
    print type(a)
    b = Base('B')
    b.func()
    print type(b)

Answer 2

您可以使用__new__，返回不同的子类：

class MyClass():
    def __new__(self, model):
        cls = {'A': ClassA, 'B': ClassB}[model]
        return object.__new__(cls)

class ClassA(MyClass):
    def func():
        print("This is ClassA.func")

class ClassB(MyClass):
    def func():
        print("This is ClassB.func")

a = MyClass('A')
a.func()

b = MyClass('B')
b.func()