Python | 根据值将元组分组到列表中

本篇文章介绍如何使用Python将元组根据其中的某个值分组到一个列表中。在实际开发中，我们有时需要按照某个条件对数据进行分组，这时就可以使用Python中的字典和列表等数据结构进行操作。

原始数据

我们首先需要定义一个包含若干元组的列表，每个元组包含两个值，一个是字符，另一个是整数。这是我们需要进行分组的原始数据。

data = [('A', 1), ('A', 2), ('B', 3), ('B', 4), ('B', 5), ('C', 6), ('C', 7), ('C', 8)]

方法一：使用setdefault函数

我们可以使用Python的setdefault函数，在遍历元组时创建字典来对数据进行分组，最终得到一个将元组值分组后的字典。这个字典中，键为元组中的字符值，值为包含该字符值的元组的列表。

result_dict = {}
for item in data:
    result_dict.setdefault(item[0], []).append(item)

print(result_dict)

得到的结果如下所示：

{
    'A': [('A', 1), ('A', 2)],
    'B': [('B', 3), ('B', 4), ('B', 5)],
    'C': [('C', 6), ('C', 7), ('C', 8)]
}

方法二：使用defaultdict函数

与上述方法类似，我们也可以使用Python的defaultdict函数来创建字典，从而实现对数据的分组。

from collections import defaultdict

result_dict = defaultdict(list)
for item in data:
    result_dict[item[0]].append(item)

print(result_dict)

结果与上述方法相同。

并行分组

如果希望在多个条件下对元组进行分组，我们可以使用Python的itertools库中的groupby函数进行并行分组。下面以将元组根据第一个和第二个元素分组为例。

import itertools

data = [('A', 1), ('A', 2), ('B', 3), ('B', 4), ('B', 5), ('C', 6), ('C', 7), ('C', 8)]

result_dict = {}
for key, group in itertools.groupby(data, lambda x: (x[0], x[1])):
    result_dict[key] = list(group)

print(result_dict)

得到的结果如下：

{('A', 1): [('A', 1)], 
 ('A', 2): [('A', 2)], 
 ('B', 3): [('B', 3)], 
 ('B', 4): [('B', 4)], 
 ('B', 5): [('B', 5)], 
 ('C', 6): [('C', 6)], 
 ('C', 7): [('C', 7)], 
 ('C', 8): [('C', 8)]}

总结

本篇文章介绍了两种使用Python将元组根据值进行分组的方法，以及使用itertools库中的groupby函数进行并行分组的方法。这些方法可以让我们快速、高效地对数据进行分组，并且易于理解和扩展。