Python 列表推导和生成器表达式

总结：当列表是我们最终需要的结果时，用列表推导式；当列表只是我们运算的中间过程时，用生成器表达式。

列表(list)是pthon中最常用的数据结构之一，有序，可变。

 

1.生成一个列表最原始的方法

 

def is_leap(year: int):

    return year%4 == 0 and (year%100 != 0 or year%400 == 0)

 

Leap_Year_List = []    #空列表 或者用 Leap_Year_List = list()

 

for item in range(1582, 2022):

    if is_leap(item):

        Leap_Year_List.append(item)  #通过append方法填充列表元素

 

2.列表推导式（List Comprehension）

当然，还有其他的方法来生成列表，用列表推导式我们可以将上面的代码简化至一行：

Leap_Year_List =[item for item in range(1582, 2022) if is_leap(item)]  #通过for循环和if条件生成列表

 

note1: range()函数本身返回的也是整数列表

note2: 列表推导式里可以用多个for循环，返回的则是笛卡尔积

 

3.生成器表达式（ Generator Expressions）

 

我们把列表表达式的[]换成()就得到了一个新的返回对象：生成器。

Leap_Year_Generator = (item for item in range(1582, 2022) if is_leap(item))

 

所以，推导式和生成器类型不一样（一个是数据结构，一个是某种迭代器）导致它们的适用场景有所不同。生成含大量元素的列表会占用极大的内存空间，这种场景下一边循环一边计算效率更高。比如计算1到1,000,000,000的平方和，我们并不需要先得到一百万个元素的列表再求和，可以直接用生成器表达式来求和：

total = sum(num * num for num in range(1, 1000000000))

 

当列表是我们最终需要的结果时，用列表推导式；当列表只是我们运算的中间过程时，用生成器表达式。

 

note1: 除生成器表达式之外，还可以通过函数中使用yield关键字来得到生成器（函数生成器）

note2: 上面有提到，生成器是一种迭代器。迭代器是可迭代对象（如列表）调用iter()函数得到的对象。除用于for循环之外，迭代器可以用next()方法不断调用并返回下一个值。

 

 

4.拓展

举个生成器的例子：读取文本文件获得版本号

def get_version():

    with open("version.txt", "r") as fi:

        result = next(it for it in fi if it.startswith("__version__")) #生成器表达式

        # txt中内容为  __version__ = "1.0.1"

    return result.split('"')[1]   #结果为1.0.1

 

另一个例子：读取文本文件返回列表

def get_install_list():

    with open("sw_list.txt", "r") as fi:

        lines = fi.readlines()  

        #txt中为多行内容

    return [line.strip("\n") for line in lines if line] #列表推导式