Python 字典过滤|极客教程

Python 字典过滤

在 Python 中，字典是一种无序、可变的数据类型，它是一种键值对的集合。有时候我们需要对字典进行一些过滤操作，只保留符合特定条件的键值对。本文将介绍如何使用 Python 中的内置函数和列表推导式来过滤字典。

1. 使用列表推导式过滤字典

列表推导式是 Python 中一种非常强大的语法，可以用来对可迭代对象进行快速、简洁的处理。我们可以利用列表推导式来过滤字典，并生成新的字典。

示例代码如下：

# 创建一个字典
my_dict = {'name': 'Alice', 'age': 25, 'country': 'USA'}

# 利用列表推导式过滤字典，只保留年龄小于30的键值对
filtered_dict = {key: value for key, value in my_dict.items() if key == 'age' or value < 30}

print(filtered_dict)

运行结果：

{'age': 25}

在上面的示例中，我们使用列表推导式对字典进行了过滤，只保留了年龄小于30的键值对。

2. 使用filter()函数过滤字典

除了列表推导式，我们还可以使用 Python 中的内置函数 filter() 来过滤字典。filter() 函数接受一个函数和一个可迭代对象作为参数，返回符合条件的元素。我们可以自定义一个函数来判断字典中的键值对是否符合条件。

示例代码如下：

# 自定义一个函数，判断字典中的值是否为字符串
def is_string(value):
    return isinstance(value, str)

# 创建一个字典
my_dict = {'name': 'Alice', 'age': 25, 'country': 'USA'}

# 使用filter()函数过滤字典，只保留值为字符串的键值对
filtered_dict = dict(filter(lambda item: is_string(item[1]), my_dict.items()))

print(filtered_dict)

运行结果：

{'name': 'Alice', 'country': 'USA'}

在上面的示例中，我们自定义了一个函数 is_string() 来判断字典中的值是否为字符串，然后使用 filter() 函数过滤字典，只保留了值为字符串的键值对。

3. 复杂条件下的字典过滤

在实际开发中，我们可能会遇到一些复杂的条件需要对字典进行过滤。这时候可以结合列表推导式和 filter() 函数来实现。

示例代码如下：

# 创建一个字典
my_dict = {'name': 'Alice', 'age': 25, 'country': 'USA', 'score': 90}

# 判断字典中的值是否为整数且大于等于80
filtered_dict = {key: value for key, value in my_dict.items() if key == 'score' or (isinstance(value, int) and value >= 80)}

print(filtered_dict)

运行结果：