Python 字典推导式的 6 个实用场景

字典推导式是 Python 中最优雅的特性之一。掌握它能让你写出更简洁、高效的代码。下面分享 6 个实际工作中常用的场景。

1. 快速转换键值对

# 传统写法
squares = {}
for i in range(5):
    squares[i] = i ** 2

# 推导式写法
squares = {i: i ** 2 for i in range(5)}
# {0: 0, 1: 1, 2: 4, 3: 9, 4: 16}

2. 过滤字典中的特定项

prices = {"apple": 1.5, "banana": 0.8, "orange": 2.0}

# 只保留价格大于 1 的水果
expensive = {k: v for k, v in prices.items() if v > 1}
# {"apple": 1.5, "orange": 2.0}

3. 交换键和值

user_ids = {"alice": 101, "bob": 102, "charlie": 103}

# 反向映射
id_to_user = {v: k for k, v in user_ids.items()}
# {101: "alice", 102: "bob", 103: "charlie"}

4. 从两个列表创建字典

keys = ["name", "age", "city"]
values = ["Alice", 25, "Beijing"]

user = dict(zip(keys, values))
# 或者用推导式
user = {k: v for k, v in zip(keys, values)}

5. 处理嵌套数据结构

users = [
    {"id": 1, "name": "Alice", "score": 85},
    {"id": 2, "name": "Bob", "score": 92},
]

# 创建 id 到名字的映射
name_map = {u["id"]: u["name"] for u in users}
# {1: "Alice", 2: "Bob"}

6. 条件赋值

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

# 奇数标记为 "odd"，偶数标记为 "even"
classification = {n: "odd" if n % 2 else "even" for n in numbers}
# {1: "odd", 2: "even", 3: "odd", 4: "even", 5: "odd"}

性能提示

字典推导式不仅代码简洁，通常也比传统循环更快，因为它是用 C 实现的。在处理大数据集时，性能差异会更明显。

注意事项

虽然推导式很强大，但不要过度嵌套。如果逻辑太复杂，拆分成多行循环更易读。

好的代码是写给人看的，顺便让机器能执行。