python __str__()、__iter__()迭代器

打印类实例调用的方法__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错误时退出循环。

  1. __iter__():迭代器,生成迭代对象时调用,返回值必须是对象自己,然后for可以循环调用next方法
  2. 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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值