Python groupby通过分隔符拆分列表

Question

我是Python(3.6)的新手,并且正在努力理解itertools groupby.我有以下包含整数的列表:

    list1 = [1, 2, 0, 2, 3, 0, 4, 5, 0]

但是列表也可能更长,并且在每对数字之后不必出现'0'.它也可以是3,4或更多的数字.我的目标是将此列表拆分为子列表,其中"0"用作分隔符,并且不会出现在任何这些子列表中.

    list2 = [[1, 2], [2, 3], [4, 5]]

这里已经解决了类似的问题: Python基于分隔符单词分割列表 答案2似乎对我有很大的帮助但不幸的是它只给了我一个TypeError.

    import itertools as it

    list1 = [1, 2, 0, 2, 3, 0, 4, 5, 0]

    list2 = [list(group) for key, group in it.groupby(list1, lambda x: x == 0) if not key]

    print(list2)

文件"H:/Python/work/ps0001/example.py",第13行,在list2 = [list(group)for key,group in it.groupby(list,lambda x:x =='0')if if not键]

TypeError:'list'对象不可调用

我很感激任何帮助,并很高兴终于了解groupby.

Answer 1

您正在检查"0"(str),但您的列表中只有0(int).此外,您使用的list是第一个列表的变量名称,这是Python中的关键字.

from itertools import groupby

list1 = [1, 2, 0, 2, 7, 3, 0, 4, 5, 0]
list2 = [list(group) for key, group in groupby(list1, lambda x: x == 0) if not key]

print(list2)

这应该给你:

[[1, 2], [2, 7, 3], [4, 5]]