迭代器与生成器

Python可迭代对象，迭代器，生成器的区别

# -*- encoding: utf-8 -*-
"""
@File    : interate_test.py
@Time    : 2020/10/16 8:56
@Author  : XUDA
@Email   : gudianpai@qq.com
@Software: PyCharm
"""
#从一个序列创建迭代器
l = [1,2,3,4,5]
# for i in l:
#     print(i)
# # ll = iter(l)
# # print(ll)
# # print(dir(ll))
# # print(ll.__next__())
# # print(ll.__next__())
# # print(ll.__next__())
# # print(ll.__next__())
# # print(ll.__next__())
# # print(ll.__next__())
l2 = iter(l)
print(next(l2))
print(next(l2))
print(next(l2))
print(next(l2))
print(next(l2))
#生成12345的平方

class PowNumber(object):
    """
    迭代器
    生成1,2,3,4,5,... 数的平方
    """
    value = 0

    def __next__(self):
        self.value += 1
        if self.value > 10:
            raise StopIteration
        return self.value * self.value

    def __iter__(self):
        return self


if __name__ == '__main__':
    pow = PowNumber()
    # print(pow.__next__())
    # print(pow.__next__())
    # print(pow.__next__())
    # print(pow.__next__())
    print(next(pow))
    print(next(pow))
    print(next(pow))
    print(next(pow))
    #循环迭代器
    for i in pow:
        print(i)

# -*- encoding: utf-8 -*-
"""
@File    : generator.py
@Time    : 2020/10/16 10:07
@Author  : XUDA
@Email   : gudianpai@qq.com
@Software: PyCharm
"""
#列表推导式
l = [x * x for x in [1,2,3,4,5]]
print(l)
l1 = (x * x for x in [1,2,3,4,5])
print(l1)
# E:\py-learn\xd_cbir\venv\Scripts\python.exe E:/py-learn/chapter09/generator.py
# [1, 4, 9, 16, 25]
# <generator object <genexpr> at 0x000002832EE1CA50>
print(l1.__next__())
print(next(l1))
# 1
# 4

# -*- encoding: utf-8 -*-
"""
@File    : generator_test.py
@Time    : 2020/10/16 10:13
@Author  : XUDA
@Email   : gudianpai@qq.com
@Software: PyCharm
"""


def pow():
    yield 1
    yield 2
    yield 3
    yield 4


def pow_number():
    return (x * x for x in [1, 2, 3, 4, 5])


def pow_number():
    for x in [1, 2, 3, 4, 10]:
        yield x * x


if __name__ == '__main__':
    rest = pow()
    print(rest)
    # #<generator object pow at 0x00000199185BCA50>
    # print(next(rest))
    # print(next(rest))
    # print(next(rest))
    # for i in rest:
    #     print(i)
    rest = pow_number()
    print(rest)
    print(next(rest))
    print(next(rest))
    print(next(rest))
    print(next(rest))
    print(next(rest))

# -*- encoding: utf-8 -*-
"""
@File    : range.py
@Time    : 2020/10/16 14:01
@Author  : XUDA
@Email   : gudianpai@qq.com
@Software: PyCharm
"""

import numpy as np


def use_range():
    """
    python内置range函数
    """
    for i in range(5, 10):
        print(i)


class InterRange(object):
    """使用迭代器来模拟range函数"""

    def __init__(self, start, end):
        self.start = start - 1
        self.end = end

    def __next__(self):
        self.start += 1
        if self.start >= self.end:
            raise StopIteration
        return self.start

    def __iter__(self):
        return self


class GenRange():
    """使用生成器来模拟range函数"""

    def __init__(self, start, end):
        self.start = start - 1
        self.end = end

    def get_num(self):
        while True:
            if self.start >= self.end - 1:
                break
            self.start += 1
            yield self.start


def get_num(start, end):
    start -= 1
    while True:
        if start >= end - 1:
            break
        start += 1
        yield start


if __name__ == '__main__':
    use_range()
    print('-' * 100)
    iter = InterRange(5, 10)
    # print(next(iter))
    l = list(iter)
    print(l)
    print('-' * 100)
    gen = GenRange(5, 10).get_num()
    # print(gen)
    # print(next(gen))
    # print(next(gen))
    # print(next(gen))
    # print(next(gen))
    # print(next(gen))
    l = list(gen)
    print(l)
    gen_f = get_num(5, 10)
    print(gen_f)
    print(list(gen_f))

#!/usr/bin/env python
# coding=utf-8
 
 
class MyList(object):            # 定义可迭代对象类
 
    def __init__(self, num):
        self.data = num          # 上边界
 
    def __iter__(self):
        return MyListIterator(self.data)  # 返回该可迭代对象的迭代器类的实例
 
 
class MyListIterator(object):    # 定义迭代器类，其是MyList可迭代对象的迭代器类
 
    def __init__(self, data):
        self.data = data         # 上边界
        self.now = 0             # 当前迭代值，初始为0
 
    def __iter__(self):
        return self              # 返回该对象的迭代器类的实例；因为自己就是迭代器，所以返回self
 
    def next(self):              # 迭代器类必须实现的方法
        while self.now < self.data:
            self.now += 1
            return self.now - 1  # 返回当前迭代值
        raise StopIteration      # 超出上边界，抛出异常
 
 
my_list = MyList(5)              # 得到一个可迭代对象
print type(my_list)              # 返回该对象的类型
 
my_list_iter = iter(my_list)     # 得到该对象的迭代器实例，iter函数在下面会详细解释
print type(my_list_iter)
 
 
for i in my_list:                # 迭代
    print i