Python 将相似值列表分组到字典中

在Python中，我们可以使用循环和条件语句、defaultdict和itertools模块的groupby方法等方法来将相似值列表分组到字典中。将它们分组在一起在分析复杂数据时非常有用。在本文中，我们将了解如何使用这些方法将相似值列表分组到字典中。

方法1：使用循环和条件语句

将相似的值统一分组并转换为字典的最简单方法之一是使用循环和条件语句。在此方法中，我们初始化一个名为groups的空字典。然后，我们遍历输入列表lst中的每个项。对于每个项，我们检查它是否已经作为键存在于groups字典中。如果是，我们将该项附加到与该键关联的列表中。否则，我们在groups字典中创建一个以该项为值的新键。

语法

list_name.append(element)

在这里，append()函数是一个列表方法，用于将一个元素添加到列表的末尾。它通过将指定的元素作为新项目来修改原始列表。

示例

在下面的示例中，输入列表包含重复的值。group_list_to_dict()函数将相似的值分组，并创建一个字典，其中键表示不同的值，与每个键关联的值是该值在原始列表中的出现次数。

def group_list_to_dict(lst):
    groups = {}
    for item in lst:
        if item in groups:
            groups[item].append(item)
        else:
            groups[item] = [item]
    return groups

输出

{1: [1, 1], 2: [2, 2], 3: [3, 3], 4: [4, 4], 5: [5]}

方法2：使用collections模块中的defaultdict

Python的collections模块提供了一个defaultdict类，可以将列表中的类似值分组为字典。defaultdict会自动为不存在的键初始化一个默认值。

语法

groups = defaultdict(list)
groups[item].append(item)

在这里，该语法使用collections模块中的defaultdict()函数创建了一个名为groups的defaultdict对象，其默认值为一个空列表。键（item）用于访问groups字典中与该键关联的列表，并将item追加到该列表中。

示例

在下面的示例中，我们从collections模块导入defaultdict类。我们使用一个默认值为空列表初始化了一个名为groups的defaultdict对象。然后，我们遍历输入列表lst中的每个项。对于每个项，我们访问groups字典中的键，并将该项追加到关联的列表中。

from collections import defaultdict

def group_list_to_dict(lst):
    groups = defaultdict(list)
    for item in lst:
        groups[item].append(item)
    return groups

my_list = [1, 2, 3, 2, 4, 1, 5, 4, 3]
result = group_list_to_dict(my_list)
print(result)

输出

defaultdict(<class 'list'>, {1: [1, 1], 2: [2, 2], 3: [3, 3], 4: [4, 4], 5: [5]})

方法3：使用itertools模块中的groupby函数

Python中的itertools模块提供了一个强大的函数groupby，它允许我们将可迭代对象中的连续相似元素进行分组。我们可以利用这个函数将列表中的值分组并转换为字典。

语法

list_name.append(element)

在这里，append() 函数将传递的元素添加到应用 append() 函数的列表中。

itertools.groupby(iterable, key=None)

在这里，可迭代的可选键作为参数传递。可迭代对象可以是任何元素的集合。它返回一个生成包含连续键和分组的元组的迭代器。关键函数确定分组条件。

示例

在下面的示例中，我们从itertools模块导入groupby函数。我们初始化一个名为groups的空字典。然后，我们遍历groupby对象，该对象返回键和具有相同值的连续项组。对于每个键和组，我们将组转换为列表，并将其分配为groups字典中该键的值。

from itertools import groupby

def group_list_to_dict(lst):
    groups = {}
    for key, group in groupby(lst):
        groups[key] = list(group)
    return groups

my_list = [1, 2, 3, 2, 4, 1, 5, 4, 3]
result = group_list_to_dict(my_list)
print(result)

输出

{1: [1], 2: [2], 3: [3], 4: [4], 5: [5]}

结论

在本文中，我们讨论了如何在Python中使用不同的方法将相似值列表分组为字典。我们讨论了使用带有条件语句的for循环、来自collections模块的defaultdict以及来自itertools模块的groupby的使用。每种方法都有其自己的优点，选择方法取决于手头的任务。