python简单生成器和迭代器

迭代器

• 迭代是Python最强大的功能之一，是访问集合元素的一种方式。
• 迭代器是一个可以记住遍历的位置的对象。
• 迭代器对象从集合的第一个元素开始访问，直到所有的元素被访问完结束。迭代器只能往前不会后退。

可迭代对象

可以返回一个迭代器的对象都可称之为可迭代对象，可迭代对象实现了_iter_方法，该方法返回一个迭代器对象。
Python中可迭代的数据类型：列表list，元组tuple，字典dict，集合set，字符串string
不可迭代的数据类型：数字
可以通过重写__iter__方法来实现对象的可迭代

# 利用模块Iterable,来检测一个对象是否可以迭代
from collections.abc import Iterable

a = ['hello', 'hi', 'good', 'yes', 'no']

for x in a:
    print(x)

for x in range(10):
    print(x)
# 可以通过for循环遍历的对象都是一个可迭代对象

# for x in 23:
#   print(x)

# 判断a是否可迭代
print(isinstance(a, Iterable))
c = 12
print(isinstance(c, Iterable))

# 重写__iter__方法，可以实现对象的可迭代
class Demo(object):
    def __iter__(self):
        pass


d = Demo()
print(isinstance(d, Iterable))

# 可迭代对象不一定可以实现for循环的遍历
for m in d:
    print(m)

自定义一个可迭代对象
可以通过next()来获取迭代对象的值

class MyIterator(object):
    def __init__(self,n):
        self.n = n
        self.current = 0

    def __next__(self):
        if self.current < self.n:
            self.current += 1
            return 100
        raise StopIteration


my_iter = MyIterator(5)
print(isinstance(my_iter, Iterator))


class Demo(object):
    def __init__(self,n):
        self.n = n

    def __iter__(self):
        return MyIterator(self.n)


d = Demo(10000)
# for m in d:
#    print(m)
print(d.__iter__().__next__())
print(d.__iter__().__next__())
print(d.__iter__().__next__())
print(d.__iter__().__next__())

简易迭代器

重写__iter__方法，一个对象就是可迭代对象，一个类要想成为迭代器，要重写__next__方法
StopIteration:用来表示一个迭代的结束，抛出Stopiteration告诉解释器，迭代完成，防止出现死循环

class MyIterable(object):
	def __init__(self,n):
		self.n = n
		self.current = 0
	def __iter__(self):
		return self
	def __next__(self):
		if self.current < self.n:
			self.current += 1
			return self.current
		raise StopIteration

my_interable = MyIterable(10)
for i in my_interable:
	print(i)
	

使用迭代器实现斐波那契数列

class Feibonacci(object):
	"""docstring for Feibonacci"""
	def __init__(self, n):
		self.n = n
		self.current = 0
		self.num1 = 1
		self.num2 = 1

	def __iter__(self):
		return self

	def __next__(self):
		if self.current < self.n:
			x = self.num1
			self.current += 1
			self.num1, self.num2 = self.num2, self.num1 + self.num2
			return x
		raise StopIteration


fb = Feibonacci(10)

list1 = []

for i in fb:
	list1.append(i)

print(list1)

生成器

• 生成器可以说就是一个函数，其和普通函数的区别在于它使用yield来返回，而不是return
• 在调用生成器运行的过程中，每次遇到 yield 时函数会暂停并保存当前所有的运行信息，返回 yield 的值, 并在下一次执行 next() 方法时从当前位置继续运行。
• 调用一个生成器函数，返回的是一个迭代器对象。
• 生成器是一个特殊的迭代器

简单的生成器实现

注意：生成器中的循环一定要使用while循环，for循环实现不了生成器

def mygeneration(n):
	current = 0
	while current < n:
		yield current
		current += 1

mg = mygeneration(10)
for i in mg:
	print(i)

生成器实现斐波那契数列

def Febonacci(n):
	current = 0
	num1 = 1
	num2 = 1
	while current < n:
		current += 1
		yield num1
		num1, num2 = num2, num1 + num2
fb = Febonacci(10)

list1 = []

for i in fb:
	list1.append(i)

print(list1)