python：魔法方法：定制序列+迭代器

定制序列

容器类型的协议：

如果说你希望定制的容器是不可变的话，你只需要定义__len__()和__getitem__()方法。        ：如元组、字符串
如果你希望定制的容器是可变的话，除了__len__()和__getitem__()方法，你还需要定义__setitem__()和__delitem__()两个方法。

容器类型的魔法方法如下：

编写一个不可改变的自定义列表，要求记录列表中每个元素被访问的次数。

class CountList:
    def __init__(self, *args):
        self.values = [x for x in args]
        self.count = {}.fromkeys(range(len(self.values)), 0)

    def __len__(self):
        return len(self.values)

    def __getitem__(self, key):
        self.count[key] += 1
        return self.values[key]

 

注意本例中按下标进行访问字典

迭代器：类似循环

iter()：内置函数
__iter__()：魔法方法
next()
__next__()

通常迭代器有字典、元组、列表

迭代器的简单使用：

string='i am a iterable'

it=iter(string)

next(it)

next(it)

没有元素了，则抛出一个StopIteration异常

for语句的工作原理模拟如下：

string='i am a iterable'

it=iter(string)

while True:

         try:

            each=next(it)

         except StopIteration:

             break

         print(each)

---

class Fibs:
    def __init__(self):
        self.a = 0
        self.b = 1
    def __iter__(self):
         return self
    def __next__(self):
        self.a, self.b = self.b, self.a + self.b
        return self.a

使用：

fibs=Fibs（）

for each in fibs:

     print(each)

我们会发现这是个四循环，明显没有出口

可以用判断each<20条件退出循环

也可以在迭代器定义的类中进行修改：

如下：

class Fibs:
    def __init__(self, n=20):
        self.a = 0
        self.b = 1
        self.n = n
    def __iter__(self):
        return self
    def __next__(self):
        self.a, self.b = self.b, self.a + self.b
        if self.a > self.n:
            raise StopIteration
        return self.a

使用：

fibs=Fibs(100)

for each in fibs:

      print(each)