租用问题

质量为本、客户为根、勇于拼搏、务实创新

< 返回租用问题列表

Python中groupby函数如何使用,python中group by用法

发布时间:2023-08-03 08:29:16

Python中groupby函数如何使用

在Python中,`groupby`函数是`itertools`模块中的函数,用于根据指定的键对可迭代对象进行分组。它返回一个迭代器,其中每一个元素是一个`(key, group)`的元组,其中`key`是分组的键,`group`是一个迭代器,包括了与该键相关联的所有元素。
以下是`groupby`函数的基本语法:
```python
itertools.groupby(iterable, key=None)
```
其中,`iterable`是可迭代对象,例如列表、字符串、元组等,`key`是一个可选的函数,用于从每一个元素中提取分组的键。如果未提供`key`函数,则默许使用元素本身作为键。
以下是一个简单的示例,演示了怎样使用`groupby`函数对列表进行分组:
```python
import itertools
data = [1, 1, 2, 3, 3, 4, 5, 5, 5]
groups = itertools.groupby(data)
for key, group in groups:
print(key, list(group))
```
输出结果为:
```
1 [1, 1]
2 [2]
3 [3, 3]
4 [4]
5 [5, 5, 5]
```
在上述示例中,列表`data`被分为了5组,每组具有相同的元素。通过遍历`groups`迭代器,我们可以获得每一个分组的键和与之相关联的元素列表。