只保留 dict 中的几个键值 - Python

在 Python 编程中，有时候我们需要从一个字典（dict）中筛选出特定的键值对。这可以通过使用 Python 的内置方法和技巧来实现。本篇文章将介绍如何只保留字典中的几个键值，并提供了相关的 Python 代码片段。

方法一：使用字典推导式

字典推导式是一种快速筛选字典的方式，可以选择性地保留需要的键值对。下面是一个示例代码片段，展示了如何使用字典推导式只保留字典中的几个键值：

# 定义原始字典
original_dict = {'key1': 'value1', 'key2': 'value2', 'key3': 'value3', 'key4': 'value4'}

# 保留的键值列表
keep_keys = ['key1', 'key3']

# 使用字典推导式筛选保留键值对
filtered_dict = {k: original_dict[k] for k in keep_keys if k in original_dict}

print(filtered_dict)

这段代码的输出结果为：

{'key1': 'value1', 'key3': 'value3'}

方法二：使用 dict comprehension

除了字典推导式，我们还可以使用 dict comprehension（字典解析式）来实现相同的功能。下面是使用 dict comprehension 只保留字典中特定键值的代码示例：

# 定义原始字典
original_dict = {'key1': 'value1', 'key2': 'value2', 'key3': 'value3', 'key4': 'value4'}

# 保留的键值列表
keep_keys = ['key1', 'key3']

# 使用 dict comprehension 筛选保留键值对
filtered_dict = {k: original_dict[k] for k in keep_keys}

print(filtered_dict)

这段代码的输出结果与上面的示例相同：

{'key1': 'value1', 'key3': 'value3'}

使用 dict comprehension 的方式更加简洁，但它对于特别大的字典可能会导致一些性能问题。

以上就是在 Python 中只保留字典中特定键值的方法。无论是使用字典推导式还是 dict comprehension，都可以轻松地实现这个功能。希望本文对你有所帮助！