### Python `itertools.groupby` 的用法与示例 `itertools.groupby` 是 Python 标准库 `itertools` 模块中的一个函数，用于将可迭代对象中的连续相同元素分组[^1]。它的基本语法如下： ```python itertools.groupby(iterable, key=None) ``` - `iterable`: 需要分组的可迭代对象。 - `key`: 一个函数（可选），用于指定分组的依据。如果未提供，则默认按连续相同的元素进行分组。 #### 示例 1：简单分组 以下是一个简单的例子，展示如何对列表中的连续相同元素进行分组： ```python from itertools import groupby data = ['a', 'a', 'b', 'b', 'a', 'c', 'c'] result = [(key, list(group)) for key, group in groupby(data)] print(result) ``` 输出结果为： ```python [('a', ['a', 'a']), ('b', ['b', 'b']), ('a', ['a']), ('c', ['c', 'c'])] ``` 上述代码展示了如何使用 `groupby` 对连续相同的元素进行分组[^2]。 #### 示例 2：带有 `key` 参数的分组 当需要根据某个条件进行分组时，可以传递一个 `key` 函数。例如，对数字列表按照奇偶性分组： ```python from itertools import groupby data = [1, 3, 5, 2, 4, 6, 7] # 按照奇偶性分组 result = [(key, list(group)) for key, group in groupby(data, key=lambda x: x % 2)] print(result) ``` 输出结果为： ```python [(1, [1, 3, 5]), (0, [2, 4, 6]), (1, [7])] ``` 这里通过 `key` 参数指定了分组规则，即按元素是否为奇数或偶数进行分组。 #### 示例 3：处理字符串 `groupby` 也可以用于字符串的分组。例如，对字符串中连续相同的字符进行分组： ```python from itertools import groupby data = "AAABBCCDDA" result = [(key, len(list(group))) for key, group in groupby(data)] print(result) ``` 输出结果为： ```python [('A', 3), ('B', 2), ('C', 2), ('D', 2), ('A', 1)] ``` 此示例展示了如何统计字符串中每个连续字符的出现次数[^2]。 #### 注意事项 - `itertools.groupby` 只会对连续相同的元素进行分组。如果需要对整个数据集进行分组（非连续），可以结合 `sorted` 函数先对数据排序再分组。例如： ```python from itertools import groupby data = [1, 3, 2, 4, 6, 5, 7] # 先排序，再分组 sorted_data = sorted(data, key=lambda x: x % 2) result = [(key, list(group)) for key, group in groupby(sorted_data, key=lambda x: x % 2)] print(result) ``` 输出结果为： ```python [(0, [2, 4, 6]), (1, [1, 3, 5, 7])] ``` 在上述代码中，先对数据进行了排序，确保相同元素连续后再进行分组。