熬最深的夜喝最劣的酒————浅谈生成器（generator）

Posted 2021-11-26 jum-bolg

测试（test）
def s():
    print("stup1")
    n = "第一步"
    yield n                   # 类似于return 但是又不同于    赖克宝，剁一下，跳一下
    print("stup2")
    yield 2
    print("stup3")
    yield 3
    print("stup4")
    yield 4
    # raise StopIteration      #  抛出异常

f = s()
print(f.__next__())
print(f.__next__())
print(f.__next__())
print(f.__next__())
# print(f.__next__())
for x in f:                   # f为迭代对象，类似于迭代器 也可用for
    print(x)
# f.__iter__() is g
# 生成器对象，实现了可迭代接口，也实现了迭代器接口

-----------------------------------------------------------------------------------------------------------


生成器实现找素数

# 素数： 只能被一和本身整除的数
class PrimeNumber(object):
    def __init__(self, start, end):
        self.start = start
        self.end = end

    def isprimenumber(self, k):
        if k < 2:
            return False
        for x in range(2, k):         # k取模依次 2-k    为0返回 Falae 否则为true
            if k % x == 0:
                print(-k)　　　　　　　　# -表示非素数
                return False
        return True

    def __iter__(self):                 # 生成器类入口
        for k in range(self.start, self.end):
            if self.isprimenumber(k):
                yield k


for t in PrimeNumber(1, 100):
    print(t)