Group Keys to Values List Using Python

Grouping keys to values is the process of categorizing data based on specific attributes or criteria. For example, imagine you have a dataset of students, and you want to group them by their grades. This grouping allows us to easily analyze and perform computations on each category separately. In Python, we can group keys to values list in various ways like using dictionaries, defaultdict, and itertools.groupby. In this article, we will understand how we can group keys to a values list using these methods.

Using Dictionaries

We can easily group keys to values using dictionaries in Python. Let's consider an example where we have a list of fruits, and we want to group them based on their colors ?

Syntax

dictionary[key].append(element)

Here, we check if the key exists in the dictionary. If it does, we append the element to the existing list. Otherwise, we create a new list with the element.

Example

In the below example, we iterate over the list of fruits and extract the color for each fruit. We then check if the color already exists in the grouped_fruits dictionary. If it does, we append the fruit to the existing list of fruits for that color. Otherwise, we create a new list with the current fruit and assign it to the corresponding color ?

fruits = [
    {"name": "Apple", "color": "Red"},
    {"name": "Banana", "color": "Yellow"},
    {"name": "Grapes", "color": "Green"},
    {"name": "Orange", "color": "Orange"},
    {"name": "Blueberry", "color": "Blue"},
    {"name": "Strawberry", "color": "Red"},
]

grouped_fruits = {}

for fruit in fruits:
    color = fruit["color"]
    if color in grouped_fruits:
        grouped_fruits[color].append(fruit)
    else:
        grouped_fruits[color] = [fruit]

print(grouped_fruits)

{'Red': [{'name': 'Apple', 'color': 'Red'}, {'name': 'Strawberry', 'color': 'Red'}], 'Yellow': [{'name': 'Banana', 'color': 'Yellow'}], 'Green': [{'name': 'Grapes', 'color': 'Green'}], 'Orange': [{'name': 'Orange', 'color': 'Orange'}], 'Blue': [{'name': 'Blueberry', 'color': 'Blue'}]}

Using defaultdict

We can implement a more effective solution using defaultdict from the Python collections module. The defaultdict automatically initializes a value for a non-existing key, saving us from the need to explicitly check and create lists ?

Syntax

from collections import defaultdict
groups = defaultdict(list)
groups[key].append(item)

Here, the defaultdict(list) creates a dictionary-like object called groups using the defaultdict class from the collections module. The groups[key].append(item) appends an item to the list associated with the key in the groups dictionary. By using square brackets [] to access the dictionary, it either retrieves the existing list if the key already exists, or creates a new empty list as the value if key is encountered for the first time.

Example

In the below example, we create a defaultdict called grouped_fruits, and its default value is set to an empty list using defaultdict(list). Now, whenever we access a non-existing key, it automatically initializes an empty list as the value. This simplifies the grouping process, as we don't need to handle key existence separately ?

from collections import defaultdict

fruits = [
    {"name": "Apple", "color": "Red"},
    {"name": "Banana", "color": "Yellow"},
    {"name": "Grapes", "color": "Green"},
    {"name": "Orange", "color": "Orange"},
    {"name": "Blueberry", "color": "Blue"},
    {"name": "Strawberry", "color": "Red"},
]

grouped_fruits = defaultdict(list)

for fruit in fruits:
    grouped_fruits[fruit["color"]].append(fruit)

print(dict(grouped_fruits))

{'Red': [{'name': 'Apple', 'color': 'Red'}, {'name': 'Strawberry', 'color': 'Red'}], 'Yellow': [{'name': 'Banana', 'color': 'Yellow'}], 'Green': [{'name': 'Grapes', 'color': 'Green'}], 'Orange': [{'name': 'Orange', 'color': 'Orange'}], 'Blue': [{'name': 'Blueberry', 'color': 'Blue'}]}

Using itertools.groupby

We can use the groupby function from the itertools module of Python to group keys to values easily. This function allows us to group elements in an iterable based on a key function ?

Syntax

itertools.groupby(iterable, key=None)

Here, the groupby() method takes an iterable as input and an optional key function. It returns an iterator that generates tuples containing consecutive keys and groups from the iterable. The key function is used to determine the grouping criterion.

Example

In the below example, we define a key function called key_function, which categorizes the numbers as "Even" or "Odd" based on their parity. We then use groupby to group the numbers based on this key function. The groupby function returns an iterator that provides the key and an iterator of the grouped elements ?

from itertools import groupby

numbers = [1, 3, 5, 2, 4, 6, 7, 9, 8, 10]

# Sort first since groupby groups consecutive elements
numbers_sorted = sorted(numbers, key=lambda x: x % 2)

grouped_numbers = {}
key_function = lambda x: "Even" if x % 2 == 0 else "Odd"

for key, group in groupby(numbers_sorted, key_function):
    grouped_numbers[key] = list(group)

print(grouped_numbers)

{'Even': [2, 4, 6, 8, 10], 'Odd': [1, 3, 5, 7, 9]}

Comparison

Conclusion

Use defaultdict for most grouping tasks as it provides clean, readable code. Use regular dictionaries when you need full control over the grouping logic, and itertools.groupby when working with consecutive elements in sorted data.