我找不到一个明确的答案。据我所知,在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
你说呢?还有别的办法吗?
这就是我如何解决一个我必须创建的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的规则。
如果你只需要__init__,使用num_holes=None作为默认值是可以的。
如果你想要多个独立的“构造函数”,你可以把它们作为类方法提供。这些方法通常称为工厂方法。在本例中,num_holes的默认值为0。
class Cheese(object):
def __init__(self, num_holes=0):
"defaults to a solid cheese"
self.number_of_holes = num_holes
@classmethod
def random(cls):
return cls(randint(0, 100))
@classmethod
def slightly_holey(cls):
return cls(randint(0, 33))
@classmethod
def very_holey(cls):
return cls(randint(66, 100))
现在创建一个这样的对象:
gouda = Cheese()
emmentaler = Cheese.random()
leerdammer = Cheese.slightly_holey()
人们肯定更喜欢已经发布的解决方案,但由于还没有人提到这个解决方案,我认为为了完整性,值得一提。
@classmethod方法可以修改为提供一个不调用默认构造函数(__init__)的替代构造函数。相反,使用__new__创建实例。
如果不能根据构造函数参数的类型选择初始化类型,并且构造函数不共享代码,则可以使用此方法。
例子:
class MyClass(set):
def __init__(self, filename):
self._value = load_from_file(filename)
@classmethod
def from_somewhere(cls, somename):
obj = cls.__new__(cls) # Does not call __init__
super(MyClass, obj).__init__() # Don't forget to call any polymorphic base class initializers
obj._value = load_from_somewhere(somename)
return obj
实际上,对于“魔幻”价值观来说,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