__init__ 中的链接函数

Question

我正在用 python 进行一些编码，但遇到了一些小问题： 我定义了几个版本的函数，并希望在创建对象时在 __init__ 函数中分配其中一个。我需要这些函数有一个self 参数。

我找到了一种方法，但我无法判断我是否正确使用了该语言。

在 my_class.py 中

class my_class(object):
    fun_1 = lambda self,val: self.attr + val
    fun_2 = lambda self,val: self.attr - val


    def __init__(self, attr, fun_version):
        self.attr = attr

        my_class.fun = my_class.__dict__['fun_' + str(fun_version)]

在 test.py 中

import my_class

my_instance_1 = my_class.my_class(attr = 2, fun_version = 1)
print(my_instance_1.fun(1))

my_instance_2 = my_class.my_class(attr = 2, fun_version = 2)
print(my_instance_2.fun(1))

结果符合预期：

3
1

还有更优雅的方法吗？我觉得在 __init__ 函数中定义类方法很尴尬。

Answer 1

实现您想要的一种简单方法是（为简单起见，将类和测试代码放在一个文件中）：

class my_class(object):
    fun_1 = lambda self,val: self.attr + val
    fun_2 = lambda self,val: self.attr - val


    def __init__(self, attr, fun_version):
        self.attr = attr

        self.fun = getattr(self, 'fun_' + str(fun_version))




my_instance_1 = my_class(attr = 2, fun_version = 1)
print(my_instance_1.fun(1))

my_instance_2 = my_class(attr = 2, fun_version = 2)
print(my_instance_2.fun(1))

#Proof that instances are independent:
print(my_instance_1.fun(1))

打印：

3
1
3

如果fun 的变体不需要实例属性（是静态方法），则可以这样编写（也可以使用lambda 以类似于第一个变体的方式编写，但这看起来很难看） ：

class my_class(object):
    @staticmethod
    def fun_1(val):
        return 1 + val

    @staticmethod
    def fun_2(val):
        return 2 + val


    def __init__(self, attr, fun_version):
        self.attr = attr

        self.fun = getattr(my_class, 'fun_' + str(fun_version))
        # or (as above):
        # self.fun = getattr(self, 'fun_' + str(fun_version))



my_instance_1 = my_class(attr = 2, fun_version = 1)
print(my_instance_1.fun(1))

my_instance_2 = my_class(attr = 2, fun_version = 2)
print(my_instance_2.fun(1))

print(my_instance_1.fun(1))

打印

2
3
2