Python 3.6 生成器

# 列表生成式

a = [x for x in range(10)]
print(a)
# [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

b = [x * 2 for x in range(10)]
print(b)
# [0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

c = [x * 2 for x in a]
print(c)
# [0, 2, 4, 6, 8, 10, 12, 14, 16, 18]

d = [x * x for x in a]
print(d)


# [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

def f(x):
    return x * x


e = [f(x) for x in a]
print(e)
# [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

# 快速获取容器内的值
li = [1, "2", "3"]
a, b, c = li
print(type(a))
# <class 'int'>
print(a)
# 1
print(type(b))
# <class 'str'>
print(b)
# 2
print(type(c))
# <class 'str'>
print(c)
# 3
print("---------------------------------------------------")
# 生成器 Part2

s = (x for x in range(10))

print(next(s))  # 0
print(next(s))  # 1
print(next(s))  # 2
print(next(s))  # 3

for i in s:
    print(i)

    # 5
    # 6
    # 7
    # 8
    # 9
    # s是一个生成器，他不会先在内存里生成所需要的值，每一次next他会生成一个值，然后取出
    # 当next的时候，他会有一个位置标记，记录当前读到哪里了，超过范围就会报错，for语句会自动读到最后一个

# 同一个对象，都会被标记位置，但是可以获取另外一个对象，位置是一个对象的属性，每个对象都会有自己的位置
s = (x for x in range(10))

print(next(s))  # 0
print(next(s))  # 1
print(next(s))  # 2
print(next(s))  # 3
print("---------------------------------------------------")


# 生成器 Part3 yield
# 这种写法也可以创建生成器，func1会返回一个生成器对象
# yield代表每一次next执行的位置，yield后面跟着的参数，是当运行到这一行yield的时候返回值，具体看代码就能明白了
def func1():
    print("OK")
    yield 10
    print("NO")
    yield "aaa"


f = func1()
next(f)
# OK
next(f)
# NO

f2 = func1()
print(next(f2))
# OK
# 10

print(next(f2))
# NO
# aaa

for i in func1():
    print(i)
    print("^^^^^")

    # OK
    # 10
    # ^^^^^
    # NO
    # aaa
    # ^^^^^

for i in func1():
    print(i)
    print("^^^^^")

    # OK
    # 10
    # ^^^^^
    # NO
    # aaa
    # ^^^^^