python之列表推导式+生成器

列表推导式

即List Comprehensions，也叫列表生成式，是python内置的非常简单却强大的可用来创建list的生成式。

它的结构是在⼀个中括号⾥包含⼀个表达式， 然后是⼀个for语句， 然后是0个或多个for或者if语句。其中的表达式可以是任意类型的对象， 返回结果是⼀个新的列表（要不然怎么叫列表推导）。

同理还有字典推导式、集合推导式。

>>> [x * x for x in range(1, 11) if x % 2 == 0]
[4, 16, 36, 64, 100]

也可以循环嵌套
>>> [m +n for m in 'ABC' for n in 'XYZ']
['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']

>>> L = ['Hello', 'World', 'IBM', 'Apple']
>>> [s.lower() for s in L]
['hello', 'world', 'ibm', 'apple']

迭代对象可以加上筛选条件
>>> [x if x % 2 == 0 else -x for x in range(1, 11)]
[-1, 2, -3, 4, -5, 6, -7, 8, -9, 10]

# 用字典推导式快速对换一个字典的键和值
{v: k for k, v in my_dict.items()}

# 集合推导式区别于列表推导式的地方是{}
squared = {x**2 for x in [1, 2, 3]}
print(squared)
# print out: {1, 4}

生成器

generator，是python中一边循环一边计算的一种机制。

生成器是一个一次生成一个值的特殊类型函数。可以将其视为可恢复函数。调用该函数将返回一个可用于生成连续 x 值的Generator，简单的说就是在函数的执行过程中，yield语句会把你需要的值返回给调用生成器的地方，然后退出函数，下一次调用生成器函数的时候又从上次中断的地方开始执行，而生成器内的所有变量参数都会被保存下来供下一次使用。

列表所有数据都存放在内存中，如果其中数据非常多，而我们又只需要访问前面几个元素，那后面的绝大多数元素占用的空间就造成了浪费。而生成器就是使得列表按照某种算法推算出来，我们能够一边循环一边计算出后续的元素，从而避免不必要的空间浪费。

创建生成器的两种方法

'''把列表生成式的[]改成()，创建一个生成器'''
>>> L = [x * x for x in range(10)]
>>> L
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
>>> g = (x * x for x in range(10))
>>> g
<generator object <genexpr> at 0x00000264E0A49430>

# 通过for循环打印generator里的值
>>> g = (x * x for x in range(4))
>>> for n in g:
...    	print(n)
...
0
1
4
9

'''定义的函数中包含yield关键字时，这个函数就是一个generator函数，调用该函数创建一个对象后，在每次调用next()的时候执行，遇到yield语句返回，再次执行时从上次中断的地方继续执行'''
def generator_function():
	for i in range(5):
		yield i

for item in generator_function():
	print(item)
'''
print out:
0
1
2
3
4
'''