使用itertools groupby创建列表列表

Question

我得到的格式为parent.id_child.id的项目列表，如1_2 。 我尝试按父ID对孩子的id进行分组，例如：从输入['1_2', '2_2', '1_1', '2_1', '1_3']我需要输出[['1','2','3'], ['1','2']] 。 我试过这个：

inputlist = ['1_2', '1_1', '2_1', '1_3', '2_2' ]
outputlist= [item.split('_') for item in inputlist]
outputlist.sort()
final = [list(group) for key, group in itertools.groupby(outputlist, lambda x: x[0])]

这正确地对元素进行分组，因为我只需要获得每个项目的第二个元素。 我怎样才能做到这一点？ 另外，我可以在单个列表理解句中完成整个事情吗？

Answer 1

使用列表理解，是的; 值本身将被传递到每个组迭代器中，因此您需要再次选择：

final = [[g[1] for g in group] for key, group in itertools.groupby(outputlist, lambda x: x[0])]

您可以通过将拆分嵌套到groupby调用中来在单个表达式中完成整个操作，但这会变得相当丑陋，即使分割为多行：

final = [
    [g[1] for g in group]
    for key, group in itertools.groupby(
        sorted(item.split('_') for item in inputlist),
        lambda x: x[0])]

您可以避免对整个输入列表进行排序，并仅使用字典对较小的组进行排序以进行分组。 根据您的ID的大小，您可能也想要以数字方式对您的ID进行排序（因为文本排序是按字典顺序完成的）：

per_parent = {}
for item in inputlist:
    parent, child = item.split('_', 1)
    per_parent.setdefault(parent, []).append(child)
final = [children for parent, children in sorted(
    per_parent.items(), key=lambda pc: int(pc[0]))]

在Python 2中，使用iteritems()而不是items()来避免构建中间列表。