Python字典setdefault方法详解1

在Python中，setdefault是字典（dict）的一个实用方法，用于简化键的存在性检查和默认值初始化。它的核心逻辑是：若键存在，返回对应的值；若不存在，则插入键并设置默认值，再返回该默认值。

---

语法

value = dict.setdefault(key, default_value)

• key: 要查找或插入的键。
• default_value（可选）: 键不存在时插入的默认值。若未提供，默认为None。
• 返回值: 键对应的值（已存在或新插入的默认值）。

---

使用场景与示例

1. 避免重复检查键的存在

假设需要统计单词频率，传统做法需先检查键是否存在：

count = {}
words = ["apple", "banana", "apple", "orange"]

for word in words:
    if word in count:
        count[word] += 1
    else:
        count[word] = 1

使用setdefault可简化为：

for word in words:
    count.setdefault(word, 0)  # 若不存在，初始化为0
    count[word] += 1

---

2. 初始化嵌套数据结构

当值为列表或字典时，setdefault能自动初始化复杂结构：

data = {}
groups = [("A", "Alice"), ("B", "Bob"), ("A", "Anna")]

for group, name in groups:
    data.setdefault(group, []).append(name)

# 结果：{'A': ['Alice', 'Anna'], 'B': ['Bob']}

---

3. 链式操作与默认值

由于setdefault返回键的值，可链式操作：

people = {}
person = "Alice"
people.setdefault(person, {}).setdefault("address", {})["city"] = "New York"

# 结果：{'Alice': {'address': {'city': 'New York'}}}

---

注意事项

1. 默认值为可变对象：
   若默认值为列表或字典等可变对象，确保每次调用时生成新对象。
   错误示例（多个键共享同一列表）：

   default_list = []
   d = {}
   d.setdefault("a", default_list).append(1)
   d.setdefault("b", default_list).append(2)
   # d = {'a': [1, 2], 'b': [1, 2]}（非预期结果）
   

   正确做法（每次传递新对象）：

   d.setdefault("a", []).append(1)
   d.setdefault("b", []).append(2)
   # d = {'a': [1], 'b': [2]}
   

2. 与get()的区别：

   • get(key, default)：仅返回值，不修改字典。
   • setdefault(key, default)：若键不存在，会插入键并设置默认值。

---

总结

• 用途：简化键的存在性检查，避免冗余代码。
• 适用场景：初始化默认值、构建嵌套结构、统计或分组数据。
• 替代方案：对于复杂需求，collections.defaultdict可能更高效。例如：

  from collections import defaultdict
  count = defaultdict(int)
  for word in words:
      count[word] += 1
  

通过合理使用setdefault，可以显著提升代码的简洁性与可读性。