PYTHON
Auto-Initialize Dictionary Values with Python's defaultdict
Prevent KeyErrors and simplify code when building dictionaries with dynamic keys. Learn how collections.defaultdict automatically initializes values with a factory function, ideal for grouping data.
from collections import defaultdict
# Example: Grouping items by category
items = [
("fruit", "apple"),
("vegetable", "carrot"),
("fruit", "banana"),
("dairy", "milk"),
("vegetable", "broccoli"),
]
# Using a regular dict would require checking if key exists
grouped_items_regular = {}
for category, item in items:
if category not in grouped_items_regular:
grouped_items_regular[category] = []
grouped_items_regular[category].append(item)
print(f"Grouped with regular dict: {grouped_items_regular}")
# Using defaultdict for cleaner code
grouped_items_defaultdict = defaultdict(list) # list is the default factory
for category, item in items:
grouped_items_defaultdict[category].append(item)
print(f"Grouped with defaultdict: {grouped_items_defaultdict}")
# Example with an int factory (e.g., counting)
word_counts = defaultdict(int)
sentence = "the quick brown fox jumps over the lazy dog"
for word in sentence.split():
word_counts[word] += 1
print(f"Word counts with defaultdict: {word_counts}")
How it works: The `collections.defaultdict` is a dictionary subclass that calls a factory function to supply missing values. Instead of raising a `KeyError` when a non-existent key is accessed, it automatically inserts the key with a default value produced by the factory. This greatly simplifies code for tasks like grouping items by category, counting occurrences, or building nested data structures without explicit key existence checks, leading to cleaner and more Pythonic solutions.