python

Python中怎么对pair数据结构进行分组

小亿
99
2024-04-23 16:16:43
栏目: 编程语言

要对pair数据结构进行分组,可以使用itertools.groupby函数。首先需要对pair数据结构进行排序,然后使用itertools.groupby函数来根据指定的条件对数据进行分组。

以下是一个示例代码:

from itertools import groupby

pairs = [(1, 'a'), (1, 'b'), (2, 'c'), (2, 'd')]

# 先对pair数据结构进行排序
pairs.sort(key=lambda x: x[0])

# 使用itertools.groupby函数对数据进行分组
grouped_pairs = {key: list(group) for key, group in groupby(pairs, lambda x: x[0])}

print(grouped_pairs)

运行以上代码将输出如下结果:

{1: [(1, 'a'), (1, 'b')], 2: [(2, 'c'), (2, 'd')]}

这样就将pair数据结构按照第一个元素进行了分组。

0
看了该问题的人还看了