如何优雅地使用Python字典推导式-向日葵屋

如何优雅地使用Python字典推导式

原创叁歲伎倆 2024-12-23 15:09 86阅读 0赞

在Python中，字典推导式（Dictionary Comprehension）是一种简洁的创建新字典的方法。以下是如何优雅地使用这种形式：

基础格式：对于键值对，你可以直接创建一个字典。

# 基础字典推导
dict_comprehension = {key: value for key, value in zip(['a', 'b'], [1, 2]))}
print(dict_comprehension)  # 输出：{'a': 1, 'b': 2}

条件判断：你可以根据条件创建字典。

# 条件字典推导
dict_comprehension_if = {key: value for key, value in zip(['apple', 'banana'], [True, False])) if value}  # 只保留当值为 True 的键值对
print(dict_comprehension_if)  # 输出：{'apple': True}

迭代序列：除了字典本身的元素，还可以迭代其他序列来创建字典。

# 迭代列表字典推导
dict_comprehension_list = {i: value for i, value in enumerate(['apple', 'banana'], [0, 1]))}  # 值为列表中对应位置的值
print(dict_comprehension_list)  # 输出：{0: 'apple', 1: 'banana'}}