python - 使用 groupby 时列表对象不可调用

Question

我正在尝试通过比较第一项来对列表列表进行分组，并且在代码运行时出现“列表对象不可调用错误”

以下是我所有的代码供参考

import code
from itertools import groupby

stuff = [[1, 'a'], [1, 'b'], [1, 'c'], [2, 'a'], [2, 'b'], [2, 'c'], [2,'d']]

gb = groupby(stuff, stuff[0])

code.interact(local=locals())

安慰

>>> gb
<itertools.groupby object at 0x000000000334C4A8>
>>> for item in gb:
...     print(item)
... 
Traceback (most recent call last):
  File "<console>", line 1, in <module>
TypeError: 'list' object is not callable

关于问题可能是什么的任何想法？提前致谢

Answer 1

您正在寻找

gb = groupby(stuff, lambda item: item[0])

第二个参数是一个key应用于每个项目的函数，应该返回组键。与排序的原理相同。