我找不到一个明确的答案。据我所知,在Python类中不能有多个__init__函数。那么如何解决这个问题呢?
假设我有一个名为Cheese的类,它具有number_of_holes属性。我怎么能有两种方法来创建奶酪对象…
其中一个需要像这样的洞的数量:帕玛森=奶酪(num_holes = 15)。
还有一个不带参数,只是随机number_of_holes属性:gouda = Cheese()。
我只能想到一种方法来做到这一点,但这似乎很笨拙:
class Cheese():
def __init__(self, num_holes = 0):
if (num_holes == 0):
# Randomize number_of_holes
else:
number_of_holes = num_holes
你说呢?还有别的办法吗?
最好的答案是上面关于默认参数的那个,但我很高兴写这个,而且它确实符合“多个构造函数”的要求。使用风险自负。
新方法怎么样?
典型的实现通过使用super(currentclass, cls)调用超类的new()方法来创建类的新实例。使用适当的参数New (cls[,…]),然后在返回它之前根据需要修改新创建的实例。”
因此,您可以让新方法通过附加适当的构造函数方法来修改类定义。
class Cheese(object):
def __new__(cls, *args, **kwargs):
obj = super(Cheese, cls).__new__(cls)
num_holes = kwargs.get('num_holes', random_holes())
if num_holes == 0:
cls.__init__ = cls.foomethod
else:
cls.__init__ = cls.barmethod
return obj
def foomethod(self, *args, **kwargs):
print "foomethod called as __init__ for Cheese"
def barmethod(self, *args, **kwargs):
print "barmethod called as __init__ for Cheese"
if __name__ == "__main__":
parm = Cheese(num_holes=5)
为什么你认为你的解决方案“笨拙”?就我个人而言,在你这样的情况下,我更喜欢一个具有默认值的构造函数,而不是多个重载构造函数(Python不支持方法重载):
def __init__(self, num_holes=None):
if num_holes is None:
# Construct a gouda
else:
# custom cheese
# common initialization
对于有很多不同构造函数的非常复杂的情况,使用不同的工厂函数可能会更干净:
@classmethod
def create_gouda(cls):
c = Cheese()
# ...
return c
@classmethod
def create_cheddar(cls):
# ...
在你的奶酪的例子中,你可能想要使用奶酪的一个Gouda子类…
这就是我如何解决一个我必须创建的YearQuarter类。我创建了一个__init__,它对各种各样的输入非常宽容。
你可以这样使用它:
>>> from datetime import date
>>> temp1 = YearQuarter(year=2017, month=12)
>>> print temp1
2017-Q4
>>> temp2 = YearQuarter(temp1)
>>> print temp2
2017-Q4
>>> temp3 = YearQuarter((2017, 6))
>>> print temp3
2017-Q2
>>> temp4 = YearQuarter(date(2017, 1, 18))
>>> print temp4
2017-Q1
>>> temp5 = YearQuarter(year=2017, quarter = 3)
>>> print temp5
2017-Q3
这就是__init__和类的其余部分的样子:
import datetime
class YearQuarter:
def __init__(self, *args, **kwargs):
if len(args) == 1:
[x] = args
if isinstance(x, datetime.date):
self._year = int(x.year)
self._quarter = (int(x.month) + 2) / 3
elif isinstance(x, tuple):
year, month = x
self._year = int(year)
month = int(month)
if 1 <= month <= 12:
self._quarter = (month + 2) / 3
else:
raise ValueError
elif isinstance(x, YearQuarter):
self._year = x._year
self._quarter = x._quarter
elif len(args) == 2:
year, month = args
self._year = int(year)
month = int(month)
if 1 <= month <= 12:
self._quarter = (month + 2) / 3
else:
raise ValueError
elif kwargs:
self._year = int(kwargs["year"])
if "quarter" in kwargs:
quarter = int(kwargs["quarter"])
if 1 <= quarter <= 4:
self._quarter = quarter
else:
raise ValueError
elif "month" in kwargs:
month = int(kwargs["month"])
if 1 <= month <= 12:
self._quarter = (month + 2) / 3
else:
raise ValueError
def __str__(self):
return '{0}-Q{1}'.format(self._year, self._quarter)
这里(引用前面的答案,文档中classmethod的纯Python版本,正如这条评论所建议的那样)是一个可以用来创建多个构造函数的装饰器。
from types import MethodType
from functools import wraps
class constructor:
def __init__(self, func):
@wraps(func)
def wrapped(cls, *args, **kwargs):
obj = cls.__new__(cls) # Create new instance but don't init
super(cls, obj).__init__() # Init any classes it inherits from
func(obj, *args, **kwargs) # Run the constructor with obj as self
return obj
self.wrapped = wrapped
def __get__(self, _, cls):
return MethodType(self.wrapped, cls) # Bind this constructor to the class
class Test:
def __init__(self, data_sequence):
""" Default constructor, initiates with data sequence """
self.data = [item ** 2 for item in data_sequence]
@constructor
def zeros(self, size):
""" Initiates with zeros """
self.data = [0 for _ in range(size)]
a = Test([1,2,3])
b = Test.zeros(100)
在某些情况下,这似乎是最干净的方法(例如,Pandas中的多个dataframe构造函数),在这些情况下,为单个构造函数提供多个可选参数将是不方便的:例如,它将需要太多参数,不可读,速度较慢或使用更多的内存。然而,正如前面的评论所指出的,在大多数情况下,通过一个带有可选参数的构造函数路由,在需要的地方添加类方法可能更符合python的规则。
实际上,对于“魔幻”价值观来说,None要好得多:
class Cheese():
def __init__(self, num_holes = None):
if num_holes is None:
...
现在,如果你想要完全自由地添加更多参数:
class Cheese():
def __init__(self, *args, **kwargs):
#args -- tuple of anonymous arguments
#kwargs -- dictionary of named arguments
self.num_holes = kwargs.get('num_holes',random_holes())
为了更好地解释*args和**kwargs的概念(实际上您可以更改这些名称):
def f(*args, **kwargs):
print 'args: ', args, ' kwargs: ', kwargs
>>> f('a')
args: ('a',) kwargs: {}
>>> f(ar='a')
args: () kwargs: {'ar': 'a'}
>>> f(1,2,param=3)
args: (1, 2) kwargs: {'param': 3}
http://docs.python.org/reference/expressions.html#calls