Python中dict.setdefault()方法的用法

发布时间：2023-05-08

在Python开发过程中，字典（dict）是经常用到的一种数据类型。对于字典的操作，Python提供了很多方法和函数。其中，dict.setdefault()方法是一项非常重要和实用的方法，能够有效地帮助我们操作字典。本文将详细介绍Python中dict.setdefault()方法的用法。

一、dict.setdefault()方法的概念

dict.setdefault()方法是Python内置字典类型（dict）的一种方法，用于在字典中创建一个新的键并设定其默认值。其用法如下：

dict.setdefault(key, default=None)

其中，key为要查找的键值，default为要设置的默认值。如果字典中已经存在该键，则返回该键对应的值；如果不存在，则将该键插入到字典中并设定一个默认值，返回该默认值。

二、dict.setdefault()方法的实例讲解

1. 参数只有键

首先，我们看一下只传入键值参数时的示例代码：

fruit = {'apple': 2, 'pear': 1, 'orange': 3}
apple_num = fruit.setdefault('apple')
print(apple_num)    # 2

在上面的代码中，我们创建了一个字典fruit，其中有3个键值对。然后，我们调用了fruit的setdefault()方法，传入了'apple'这个键值。由于fruit中存在'apple'这个键值，因此方法返回的是该键值对应的值（也就是2）。在这个过程中，方法并没有修改键值对。

2. 参数为键和默认值

那么，如果我们传入默认值呢？请看下面的代码：

fruit = {'apple': 2, 'pear': 1, 'orange': 3}
banana_num = fruit.setdefault('banana', 4)
print(fruit)        # {'apple': 2, 'pear': 1, 'orange': 3, 'banana': 4}
print(banana_num)   # 4

在上面的代码中，我们调用了fruit的setdefault()方法，传入了'banana'这个键值和4作为默认值。由于fruit中不存在'banana'这个键值，方法将该键插入到字典中并设定默认值4，返回的值也是4。这样，fruit中就多了一个键值对{'banana': 4}。

3. 参数只有默认值

那么，如果我们只传入默认值，而不传入键值呢？请看下面的代码：

fruit = {'apple': 2, 'pear': 1, 'orange': 3}
num = fruit.setdefault('banana')
print(fruit)    # {'apple': 2, 'pear': 1, 'orange': 3, 'banana': None}
print(num)      # None

在上面的代码中，我们调用了fruit的setdefault()方法，不传入键值，只传入默认值。由于fruit中不存在该键值，方法会将该键插入到字典中并设定默认值为None。然后，方法返回值也是None。

4. 字典类型的嵌套使用

接下来，我们看一下dict.setdefault()方法在字典类型嵌套使用时的案例。请看下面的代码：

fruit = {'apple': {'red': 1, 'green': 1}, 'pear': {'yellow': 1}}
num = fruit.setdefault('orange', {'big': 1, 'small': 2})
print(fruit)    # {'apple': {'red': 1, 'green': 1}, 'pear': {'yellow': 1}, 'orange': {'big': 1, 'small': 2}}
print(num)      # {'big': 1, 'small': 2}

在上面的代码中，我们创建了一个字典fruit，其中包含两个键值对。然后，我们调用了fruit的setdefault()方法，传入了'orange'这个键值和{'big': 1, 'small': 2}作为默认值。由于fruit中不存在'orange'这个键值，方法将该键插入到字典中并设定默认值{'big': 1, 'small': 2}。然后，方法返回该默认值。这样，fruit中就多了一个键值对{'orange': {'big': 1, 'small': 2}}。

三、dict.setdefault()方法的应用场景

dict.setdefault()方法在Python中是一个非常重要和实用的方法，它可以帮助我们在字典操作过程中更加方便地创建新键值，并设定默认值。以下是dict.setdefault()方法的一些实际应用场景：

1. 遍历列表生成字典

# 统计水果的数量并生成字典
fruit_name = ['apple', 'banana', 'pear', 'orange']
fruit_num = [2, 4, 1, 3]
fruit_dict = {}
for i in range(len(fruit_name)):
    fruit_dict.setdefault(fruit_name[i], fruit_num[i])
print(fruit_dict)    # {'apple': 2, 'banana': 4, 'pear': 1, 'orange': 3}

2. 处理信息获取字典中嵌套数据

# 统计所有同学的年龄信息
stu_info = {'Pony': {'age': 23, 'gender': 'male'},
            'Lucas': {'age': 21, 'gender': 'male'},
            'Lisa': {'gender': 'female'}}
for name in stu_info.keys():
    stu_info[name].setdefault('age', 0)
print(stu_info)    # {'Pony': {'age': 23, 'gender': 'male'}, 'Lucas': {'age': 21, 'gender': 'male'}, 'Lisa': {'gender': 'female', 'age': 0}}

3. 在爬虫中进行数据处理的优化

import requests
import json
def spider():
    response = requests.get('http://xxx.xx.xx')
    content = json.loads(response.text)
    data = {}
    for item in content['data']:
        type = item['type']
        data.setdefault(type, [])
        data[type].append(item)
    return data

在上面的代码中，spider()函数是一个爬虫函数，用于爬取数据并进行处理。通过setdefault()方法，我们可以在执行循环时动态地创建键并添加数据，这样可以更加高效地对数据进行处理。

总结

本文详细介绍了Python中dict.setdefault()方法的用法。通过对该方法的实际讲解和实际应用场景的说明，相信读者已经对该方法有了更深入的理解。在实际Python开发过程中，应该灵活运用dict.setdefault()这一方法，帮助完成各种不同的数据操作任务。

推荐文章