要對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數據結構按照第一個元素進行了分組。