python 列表生成式和生成器

列表生成式和生成器

1、列表普通定义

列表普通定义如下：

list1 = [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]  # 普通自定义

list1 = list('123456789')
# ['1', '2', '3', '4', '5', '6', '7', '8', '9']

这样的写法比较繁琐，元素较少还行，如果需要生成多个元素的列表，那么一个一个的写入到列表中，这样的效率显然很慢，所以不可取。

2、列表生成式

列表生成式：会将所有的结果全部计算出来，把结果存放到内存中，如果列表中数据比较多，会占用过多的内存空间，可能会导致MemoryError内存错误或者导致程序在运行时出现卡顿的情况。

list1 = []
for x in range(10):
    list1.append(x * x)
print(list1)
# 结果: 
# [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

以上列表生成式也可以写成：

list1 = [x * x for x in range(10)]

输出的结果是等效的，[x * x for x in range(10)]的写法更简洁。列表生成式还可以与if条件语句搭配使用，比如生成0到100之间的偶数的列表。

list2 = [x for x in range(50) if x % 2 == 0]

3、列表生成器

列表生成器：生成器是一边循环一边计算的机制，会创建一个列表生成器对象，不会一次性的把所有结果都计算出来，如果序号获取数据，可以使用next()函数来获取，也可使用for循环遍历。
注意：next()函数获取不到数据，会导致出现StopIteration异常错误，推荐使用for循环遍历生成器对象，获取所有数据。

list1 = []
gen = (x * x for x in range(10))
for num in gen:
    list1.append(num)
print(list1)

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

同样的列表生成器中也可以与if条件语句搭配使用，以下就是输出[0，100]之间能够被5整除的生成器写法。

list1 = []
gen = (x for x in range(101) if x % 5 == 0)
for num in gen:
    list1.append(num)
print(list1)

# [0, 5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70, 75, 80, 85, 90, 95, 100]

列表生成式和生成器犹如一个时间与空间的矛盾体。列表生成式将所有结果计算出，存储在内存中，消耗了计算机的内存，却在使用过程中节省了时间。然而生成器不会把所有结果计算出来，节省了内存，却在使用数据的时候，需要遍历取出数据，消耗了更多的时间，这点与列表生成式恰好相反。