python迭代器、生成器与装饰器

迭代器

迭代器简单理解为一个容器,循环的时候每次从容器中取出一个数据,直到容器数据取完。

自定义迭代器,需要在类中实现两个方法__iter__和__next__

__iter__:方法需要返回对象本身,它是for循环使用迭代器的要求

__next__:方法用于返回容器中下一个元素,当容器中的数据取完时,需要引发StopIteration异常。

class Number():

    def __init__(self,min,max):

        self.min=min

        self.max=max

    def __iter__(self):

        return self

    def __next__(self):

        num =self.min

        if self.min<=self.max:

            self.min +=1

            return num **3

        else:

           raise StopIteration

num_list=[]

for iin Number(1,6):

    print(i)

    num_list.append(i)

print(num_list)

上面这段代码简单写了一下自定义迭代器,先定义了一个Number类,初始化方法里有min和max两个参数,定义两个方法__iter__和__next__,然后将min赋值给一个变量num,判断min是否<=max,如果小于等于返回num的3次方。然后抛出异常,最后用for循环测试1-6之间的3次方,将值添加到列表中。注意self.min +=1,不加会死循环。

生成器

假如有一个列表,里面有很多元素,但是只想访问其中的某几个元素,这样的操作会特别耗内存。

生成器的特点:

1、可以操作海量数据,节约大量内存

2、跟普通函数不同,生成器是一个返回迭代器的函数,只能用于迭代操作,更简单的理解生成器就是一个迭代器

3、在调用生成器运行的过程中,每次遇到yield时,函数会暂停并保存当前运行的所有信息,返回yield的值,并在下一次执行next()方法时从当前位置继续运行。

yield和return的区别

1、return会跳出当前函数,yield不会跳出当前函数

2、yield保存当前函数的执行状态,在返回后,函数又回到之前保存的状态继续执行

简单说一下闭包

闭包的特点:

1、函数内部,再定义函数

2、内部函数引用外部变量,但并非全局变量

3、需要把内部函数返回

def outer():

    a=0

    def inner(b):

        #print("")

        print(a+b)

        return inner

f=outer()

f.__closure__[0].cell_contents

上述代码简单实现了一个闭包,在outter函数内有一个变量a和一个inner函数,inner函数引用了变量a,并返回inner,最后使用f.__closure__[0].cell_contents代码进行检测,如果没有报错那说明这个outter函数是闭包,如果报错则不是闭包。将print("")注释取消,注释print(a+b),测试运行会报错,闭包必须满足特点中的3个条件。

装饰器

装饰器是一种增加函数或类功能的简单方法,它可以快速的给不同的函数或类传入相同的功能,调用被装饰过的函数与之前调用函数并没有区别。

装饰器基本格式:

def decorate(func):  #定义装饰器函数,参数为func,代表被装饰的函数

    def wrapper(*args,**kwargs):  #新定义一个包装函数,用于返回

        func(*args,**kwargs)  #调用被装饰的函数

    return wrapper  #返回包装函数

装饰器的简单用法:

@decorate

def func():

    print("hello func")

def func1():

    print("hello func1")

def func2():

    print("hello func2")

#定义装饰器

def decorate(func):

    def wrapper(*args,**kwargs):

        print("开始")

        func(*args,**kwargs)

        print("结束")

    return wrapper

func()

func1()

func2()

只需要在调用被装饰的函数上面或者下面添加需要的代码,然后再原函数上用@装饰器名正常调用函数就可以了。

你可能感兴趣的:(python迭代器、生成器与装饰器)