Python 中groupby
时间: 2023-07-30 16:12:44 浏览: 72
详解python中groupby函数通俗易懂
3星 · 编辑精心推荐
groupby 是 Python 中的一个函数,它可以对一个可迭代对象进行分组操作。具体使用方法如下:
1. 首先需要导入 itertools 模块:`import itertools`
2. 然后可以使用 groupby 函数进行分组操作,语法如下:
`groupby(iterable, key=None)`
其中 iterable 为要分组的可迭代对象,key 为分组的关键字,如果没有指定,则默认使用元素本身进行分组。
3. 最后返回的是一个迭代器,可以通过循环来遍历分组后的结果。
示例代码如下:
```python
import itertools
lst = [1, 2, 3, 4, 5, 6, 7, 8, 9]
for key, group in itertools.groupby(lst, lambda x: x % 2):
print(key, list(group))
```
输出结果为:
```
1 [1, 3, 5, 7, 9]
0 [2, 4, 6, 8]
```
以上代码将列表 lst 中的奇数和偶数分别分组输出。其中 lambda x: x % 2 为自定义的分组关键字,对于 lst 中的每个元素,如果它的值模 2 后为 1,那么它就是奇数,否则就是偶数。
阅读全文