python中groupby函数用法
时间: 2023-11-09 22:48:27 浏览: 121
详解python中groupby函数通俗易懂
3星 · 编辑精心推荐
在Python中,`groupby`函数是`itertools`模块中的一个函数。它用于按照指定的条件对可迭代对象进行分组。
下面是`groupby`函数的基本用法:
```python
from itertools import groupby
# 定义一个可迭代对象
data = [1, 1, 2, 3, 3, 3, 4, 5, 5]
# 使用groupby函数进行分组
groups = groupby(data)
# 遍历分组结果
for key, group in groups:
print(key, list(group))
```
输出结果为:
```
1 [1, 1]
2 [2]
3 [3, 3, 3]
4 [4]
5 [5, 5]
```
在上面的例子中,我们定义了一个可迭代对象`data`,其中包含一些整数。通过调用`groupby(data)`,我们将`data`按照连续相等的元素进行分组。然后,我们使用`for`循环遍历分组结果,每个分组结果返回一个键(即相同元素的值)和一个包含相同元素的迭代器。
需要注意的是,`groupby`函数要求可迭代对象已经按照分组的条件进行排序。如果未排序,则需要事先对数据进行排序,例如使用`sorted(data)`来对`data`进行排序。
阅读全文