python .groupby分组函数的用法
时间: 2024-05-08 20:12:20 浏览: 102
Python中的groupby分组功能的实例代码
Python中的`.groupby()`函数是用于对可迭代对象进行分组的函数。它根据指定的键函数对元素进行分组,并返回一个由分组后的元素组成的迭代器。
使用`.groupby()`函数需要先导入`itertools`模块,然后调用`groupby()`函数并传入两个参数:可迭代对象和键函数。键函数用于指定分组的依据,可以是一个lambda表达式或者自定义的函数。
下面是一个示例代码,演示了`.groupby()`函数的用法:
```python
import itertools
# 定义一个列表
data = [1, 1, 2, 3, 3, 4, 5, 5]
# 使用groupby函数进行分组
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]
```
在上述示例中,我们定义了一个列表`data`,其中包含了一些整数。然后我们使用`.groupby()`函数对`data`进行分组,并通过遍历分组后的结果来打印每个分组的键和对应的元素列表。
阅读全文