Python推导式详细讲解

更多资料获取

📚 个人网站：ipengtao.com

在Python中，推导式是一种简洁而强大的语法特性，可以用来创建列表、集合、字典等数据结构。本文将深入探讨Python中的三种推导式：列表推导式、集合推导式和字典推导式，以及它们在不同情境下的应用。

列表推导式

列表推导式是Python中最常见的推导式，它允许我们用一行代码创建新的列表。

以下是一些实例：

# 生成1到10的平方列表
squares = [x**2 for x in range(1, 11)]

# 过滤偶数，生成仅包含奇数的列表
odd_numbers = [x for x in range(1, 11) if x % 2 != 0]

# 使用条件表达式生成包含奇偶性判断的列表
evenodd_label = ["Even" if x % 2 == 0 else "Odd" for x in range(1, 11)]

集合推导式

集合推导式类似于列表推导式，但生成的是集合。

以下是一些集合推导式的示例：

# 生成1到10的平方集合
squares_set = {x**2 for x in range(1, 11)}

# 过滤偶数，生成仅包含奇数的集合
odd_numbers_set = {x for x in range(1, 11) if x % 2 != 0}

字典推导式

字典推导式可以用一行代码生成字典，非常适用于从其他数据结构构建字典的场景。

# 生成1到10的数字与其平方的字典
squares_dict = {x: x**2 for x in range(1, 11)}

# 过滤偶数，生成仅包含奇数的字典
odd_numbers_dict = {x: "Odd" for x in range(1, 11) if x % 2 != 0}

推导式的性能

在Python中，推导式是一种优雅而简洁的语法特性，但在使用时需要注意其性能，特别是在处理大规模数据时。我们将对比推导式与传统循环在性能上的差异，并讨论何时选择使用推导式以及何时选择传统循环。

1. 推导式 vs. 传统循环

考虑以下示例，使用列表推导式和传统循环分别生成包含1到1000的平方的列表：

列表推导式:

squares = [x**2 for x in range(1, 1001)]

传统循环:

squares = []
for x in range(1, 1001):
    squares.append(x**2)

2. 性能比较

为了比较性能，我们可以使用Python内置的timeit模块来测量代码的执行时间。以下是一个简单的性能测试：

import timeit

# 列表推导式性能测试
time_list_comprehension = timeit.timeit('[x**2 for x in range(1, 1001)]', number=1000)

# 传统循环性能测试
time_for_loop = timeit.timeit("""
squares = []
for x in range(1, 1001):
    squares.append(x**2)
""", number=1000)

print(f"列表推导式执行时间: {time_list_comprehension} 秒")
print(f"传统循环执行时间: {time_for_loop} 秒")

3. 何时选择推导式？

推导式在简化代码和提高可读性方面具有优势，但在性能上可能不总是最佳选择。推导式通常在处理简单任务和小型数据集时表现良好。当任务变得复杂或数据规模庞大时，传统循环可能更具有灵活性和控制性。

选择使用推导式的情况：

任务简单，代码清晰易读。
处理小型数据集，性能影响可接受。

4. 何时选择传统循环？

传统循环在处理复杂任务和大规模数据集时可能更为适用。它提供更多的灵活性和可控性，有时可以通过手动优化提高性能。

选择使用传统循环的情况：

任务复杂，需要更多控制结构。
处理大型数据集，性能至关重要。

嵌套推导式

嵌套推导式是Python中强大的特性之一，它允许在一个推导式内部包含另一个推导式，用于处理多维数据结构，例如二维列表或字典。本节将深入讨论嵌套推导式的用法以及在处理多维数据时的应用场景。

1. 二维列表的创建

考虑一个场景，想要创建一个5x5的九九乘法表。嵌套推导式能够以紧凑而清晰的方式实现这个任务：

multiplication_table = [[i * j for j in range(1, 6)] for i in range(1, 6)]
print(multiplication_table)

这个嵌套推导式创建了一个包含九九乘法表的二维列表，其中外层推导式负责生成每一行，内层推导式负责生成每一列的元素。这种结构使得代码易于理解且高度可读。

2. 过滤嵌套列表

嵌套推导式也可以用于过滤多维数据结构。例如，想要获取九九乘法表中所有元素值为偶数的项：

even_values = [value for row in multiplication_table for value in row if value % 2 == 0]
print(even_values)

这个例子中，通过在嵌套推导式中添加条件语句，可以轻松地筛选出符合条件的元素。

3. 字典的嵌套推导式

嵌套推导式同样适用于字典的创建和过滤。考虑一个示例，想要创建一个包含坐标点的字典，其中x和y的取值范围在1到3之间：

coordinate_dict = {(x, y): x * y for x in range(1, 4) for y in range(1, 4)}
print(coordinate_dict)

这个嵌套推导式创建了一个字典，其中包含了所有可能的坐标点及其对应的乘积。

4. 多重条件的嵌套推导式

在嵌套推导式中，可以使用多重条件来进一步筛选数据。例如，如果只想要九九乘法表中那些乘积大于10的元素：

filtered_values = [value for row in multiplication_table for value in row if value > 10]
print(filtered_values)

通过添加条件语句，我们可以方便地实现对多维数据结构的复杂过滤操作。

推导式的应用场景

推导式是Python中一项强大而灵活的语法特性，适用于各种数据处理、过滤和转换场景。以下是几个实际案例，展示推导式在不同应用场景中的优雅应用。

1. 数据初始化

在某些情况下，需要初始化一个包含特定值的列表。使用列表推导式可以使这个过程非常简洁：

# 初始化一个包含5个零的列表
zero_list = [0 for _ in range(5)]
print(zero_list)

2. 过滤与转换

假设有一个包含数字的列表，想要生成一个新列表，其中包含原列表中每个元素的平方值，但仅保留偶数的平方值：

original_list = [1, 2, 3, 4, 5]

# 使用推导式过滤与转换
squares_of_evens = [x**2 for x in original_list if x % 2 == 0]
print(squares_of_evens)

3. 字典构建

字典推导式可以用于从其他数据结构构建字典。考虑一个场景，有两个列表，一个表示城市，另一个表示对应的人口数量：

cities = ['New York', 'Tokyo', 'London']
populations = [8537673, 37977073, 9304016]

# 使用字典推导式构建城市与人口的字典
city_population_dict = {city: population for city, population in zip(cities, populations)}
print(city_population_dict)

4. 文件读取与处理

在处理文件时，推导式可以快速生成某些数据的列表。例如，读取一个文件中的每一行，提取所有大写单词：

with open('sample.txt', 'r') as file:
    uppercase_words = [word for line in file for word in line.split() if word.isupper()]
print(uppercase_words)

5. 多维数据处理

使用嵌套推导式可以轻松处理多维数据结构，例如创建一个包含矩阵转置的列表：

matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

# 使用嵌套推导式进行矩阵转置
transposed_matrix = [[row[i] for row in matrix] for i in range(len(matrix[0]))]
print(transposed_matrix)