python3 生成器表达式

2023-05-01 20:29 由云淡#风清发表于 #后端开发

在 Python3 中，生成器表达式是一种语言结构，它可以快速地创建一个可迭代对象。生成器表达式类似于列表推导式，但使用圆括号而不是方括号，并且返回的是一个生成器对象而不是一个列表。

在 Python3 中，生成器表达式有两种类型：生成器函数和生成器表达式。

生成器函数：

生成器函数是一种特殊的函数，在函数中使用 yield 语句来生成一个值，然后暂停函数执行并保留当前状态，等待下一次调用时继续执行。生成器函数的优点是可以处理大量数据，因为它们只需要在内存中保存一个值，而不是全部保存在内存中。

例如，以下是一个生成器函数，它可以生成斐波那契数列中的前 n 个数字：

def fibonacci(n):
    a, b = 0, 1
    for i in range(n):
        yield a
        a, b = b, a + b

生成器表达式：

生成器表达式是使用圆括号包围的表达式，其中包含一个 for 循环和一个可选的 if 条件。生成器表达式可以用来生成一个序列，这个序列可以通过迭代访问，但不必事先将所有元素保存在内存中。

例如，以下生成器表达式可以生成一个包含从 1 到 10 的偶数的生成器对象：

gen = (i for i in range(1, 11) if i % 2 == 0)

注意事项：

生成器表达式可以节省内存空间，但是如果需要多次使用生成器对象中的值，则需要将其转换为列表或其他数据结构。
如果生成器表达式中的代码太长或复杂，则建议使用生成器函数来代替，以提高代码的可读性和可维护性。
如果生成器表达式中的代码有副作用（例如修改了全局变量），则可能会导致意外行为，应该避免这种情况。
生成器表达式可以嵌套，但是应该注意不要嵌套过深导致代码难以理解。例如：
```
gen = ((i, j) for i in range(1, 4) for j in range(4, 7))
```
这个生成器表达式可以生成一个包含所有 (1,4) 到 (3,6) 的元组的生成器对象。
生成器表达式中的 for 循环可以有多个，每个循环可以使用一个 if 条件。例如：
```
gen = (i * j for i in range(1, 4) if i % 2 == 0 for j in range(4, 7) if j % 2 != 0)
```
这个生成器表达式可以生成一个包含所有偶数 i 与奇数 j 的乘积的生成器对象。

生成器表达式中的变量作用域只在生成器表达式内部，不会泄露到外部。例如：

x = 10
gen = (x for x in range(1, 5))
print(list(gen))    # 输出 [1, 2, 3, 4]
print(x)    # 输出 10，说明 x 只在生成器表达式内部存在，不会影响外部变量 x 的值。

生成器表达式可以和其他 Python 的内置函数或模块一起使用，例如 map、filter、itertools 等。例如：

import itertools

# 使用 map 函数和生成器表达式生成一个列表，其中每个元素都是平方数。
lst = list(map(lambda x: x ** 2, (i for i in range(1, 5))))
print(lst)    # 输出 [1, 4, 9, 16]

# 使用 itertools 模块中的 zip_longest 函数和生成器表达式生成一个包含所有输入迭代器的元组的列表。
lst = list(itertools.zip_longest((i for i in range(1, 5)), ('a', 'b', 'c')))
print(lst)    # 输出 [(1, 'a'), (2, 'b'), (3, 'c'), (4, None)]

在使用生成器表达式时，应该尽可能地使用惰性求值，即只生成需要的元素，并且在使用完之后立即释放相应的资源。这样可以避免不必要的内存占用和性能问题。
处理大型数据集，例如从文件或数据库中读取数据，并将其用作生成器表达式的输入。这样可以避免一次性加载所有数据，并且节省内存空间。
```
with open('data.txt') as f:
    gen = (line.strip() for line in f if 'error' in line)
    for item in gen:
        print(item)
```
通过生成器表达式实现惰性求值，例如只有当需要时才计算函数的值。这样可以避免不必要的计算和内存占用。
```
def expensive_function(n):
    print(f"Calculating {n}...")
    return n ** 2

gen = (expensive_function(i) for i in range(5))
print(list(gen))    # 输出 Calculating 0... Calculating 1... Calculating 2... Calculating 3... Calculating 4... [0, 1, 4, 9, 16]
```
这个例子中，我们定义了一个函数 expensive_function，并使用一个生成器表达式来生成一个包含前五个数字的平方的列表。在评估生成器表达式时，expensive_function 只有在需要计算平方时才被调用，这样可以避免不必要的计算和内存占用。

在多个迭代器之间生成元素，例如合并两个排序列表并返回一个新的排序列表。

def merge_sorted(lst1, lst2):
    i, j = 0, 0
    while i < len(lst1) and j < len(lst2):
        if lst1[i] <= lst2[j]:
            yield lst1[i]
            i += 1
        else:
            yield lst2[j]
            j += 1
    yield from lst1[i:]
    yield from lst2[j:]

lst1 = [1, 3, 5, 7]
lst2 = [2, 4, 6, 8]
gen = merge_sorted(lst1, lst2)
print(list(gen))    # 输出 [1, 2, 3, 4, 5, 6, 7, 8]

这个例子中，我们定义了一个 merge_sorted 函数来合并两个排序列表，并返回一个新的排序列表。在函数中，我们使用一个生成器函数来生成所有排好序的元素，并在函数返回之前返回它们。这个方法可以在处理大型数据集时节省内存空间，并且可以避免不必要的排序或其他操作。

用于过滤和转换数据，例如将一个列表中的所有元素转换为字符串并删除其中的空格。
```
lst = [' hello ', ' world', '', 'python', '']
gen = (s.strip() for s in lst if s)
print(list(gen))    # 输出 ['hello', 'world', 'python']
```
这个例子中，我们使用一个生成器表达式来对列表中的所有元素进行过滤和转换。具体来说，我们首先使用 if 子句来过滤出所有不为空的字符串，然后使用 strip 方法来删除每个字符串的前导和尾随空格。最后，我们将经过处理的字符串返回为一个生成器对象，并将其转换为一个列表。
生成器表达式可以与其他 Python 内置函数（如 map 和 filter）和模块（如 itertools）结合使用，以实现更高效和优雅的代码。
在使用生成器表达式时，应该尽可能地使用惰性求值，即只生成需要的元素，并在使用完之后立即释放相应的资源。这样可以避免不必要的内存占用和性能问题。
如果生成器表达式中的代码有副作用（例如修改了全局变量），则可能会导致意外行为，应该避免这种情况。
在编写长的生成器表达式时，建议将其分解成多个简单的表达式或生成器函数，以提高代码的可读性和可维护性。
在使用生成器表达式时，应该学会使用列表推导式和普通的 for 循环来进行比较，以选择最适合特定任务的工具。
最后，需要注意的是，生成器表达式虽然非常强大和方便，但也并非万能的。在某些情况下，还是需要使用其他语言结构或算法来解决问题。
在使用生成器表达式时，应该避免使用过多的嵌套循环和条件语句，以免代码变得难以阅读和维护。在这种情况下，建议考虑使用其他数据结构或算法。
使用生成器表达式时，应该尽可能地保持代码简单和易读。这包括命名变量、注释代码和格式化输出，以便其他人可以理解你的代码。
在使用生成器表达式时，需要注意一些性能问题。例如，在处理大型数据集时，可能需要考虑使用并行计算或其他优化算法来提高效率。让我们来看一个例子，说明生成器表达式和列表推导式之间的差异：

让我们来看一个例子，说明生成器表达式和列表推导式之间的差异：

# 列表推导式
lst = [i ** 2 for i in range(1, 11)]
print(lst)

# 生成器表达式
gen = (i ** 2 for i in range(1, 11))
print(list(gen))

个例子中，我们首先使用列表推导式创建一个包含前 10 个数字的平方的列表。然后，我们使用一个生成器表达式来创建一个包含相同元素的生成器对象，并将其转换为一个列表。

一般来说，列表推导式比起生成器表达式更加适合小型数据集，因为它们可以在内存中完全构建出一个列表，并且可以在需要时随时进行索引和修改。另一方面，生成器表达式更适合大型数据集，因为它们只返回必要的元素，而且可以逐个处理每个元素，从而节省内存空间。

在实际编程中，我们应该根据具体情况选择最适合的工具，并权衡各种因素（例如代码的可读性、效率、内存占用等）。