打印类实例调用的方法__str__ |
对于用户自定义的类,打印其实例的时候,输出的格式是该类所属的模块+类实例的内存地址。当然python也提供了一个__str__方法,用户只需要重写这个方法,就能够定制对类实例的打印输出。
>>> class Student(object):
... def __init__(self, name):
... self.name = name
...
>>> print Student('Michael')
<__main__.Student object at 0x109afb190>
下面重写__str__方法,再次看打印类实例的输出内容:
>>> class Student(object):
... def __init__(self, name):
... self.name = name
... def __str__(self):
... return 'Student object (name: %s)' % self.name
...
>>> print Student('Michael')
Student object (name: Michael)
但是,直接输出类的实例,不用print函数,打印出来的类实例仍然有问题:
>>> s = Student('Michael')
>>> s
<__main__.Student object at 0x109afb310>
这是因为直接显示变量调用的不是__str__(),而是__repr__(),两者的区别是__str__()返回用户看到的字符串,而__repr__()返回程序开发者看到的字符串,也就是说,__repr__()是为调试服务的。
解决办法是再定义一个__repr__()。但是通常__str__()和__repr__()代码都是一样的,所以,有个偷懒的写法:
class Student(object):
def __init__(self, name):
self.name = name
def __str__(self):
return 'Student object (name=%s)' % self.name
__repr__ = __str__
类实例用于返回一个迭代对象时,调用__iter__ |
如果一个类被用于for ... in
循环,类似list或者tuple,就必须实现一个__iter__()
方法,该方法返回一个迭代对象。然后Python的for循环就会不断调用该对象的next()
方法拿到下一个值,直到遇到StopIteration错误时退出循环。
- __iter__():迭代器,生成迭代对象时调用,返回值必须是对象自己,然后for可以循环调用next方法
- next():每一次for循环都调用该方法(必须存在)
class IterInstance(object):
"""
"""
def __init__(self, num):
self.num = num
self.start_num = -1
def __iter__(self):
"""
迭代器,生成迭代对象时调用,返回值必须是一个迭代对象
一般是对象自己,然后for可以循环调用这个可迭代对象的next方法
"""
print '__iter__'
return self
def next(self):
'''
每一次for循环都调用该方法(必须存在)
'''
self.start_num += 1
if self.start_num >= self.num:
raise StopIteration()
return self.start_num
if __name__ == "__main__":
for i in IterInstance(10):
print i