在Python中将数字列表分组在一起

Question

我有以下内容，这是Python中列表的列表，并且是我拥有的部分值列表：

[1,33]
[2,10,42]
[5,1,33,44]
[10,42,98]
[44,12,100,124]

有没有一种将它们分组的方法，以便它们收集每个列表中共有的值？

例如，如果我查看第一个列表[1,33]，则可以看到该值存在于第三个列表中：[5,1,33,44]

因此，这些被分组为

[5,1,33,44]

如果继续看下去，我可以看到最终列表中有44个，因此它将与该列表一起分组。

将[44,12,100,124]添加到[5,1,33,44]

给出：[1,5,12,33,44,100,124]

第二个列表[2,10,42]与[10,42,98]具有共同的值，因此将它们连接在一起可得出：[2,10,42,98]

因此，最终列表为：

[1,5,12,33,44,100,124]

[2,10,42,98]

我猜这种类型的分组有一个特定的名称。 是否有可以自动处理的库？ 还是我必须编写手动搜索方式？

我希望所做的编辑可以使我更清楚地了解要实现的目标。

谢谢。

Answer 1

这是一个不需要标准库或3rd party软件包的解决方案。 请注意，这会改变a 。 为避免这种情况，只需复制a并对其进行处理。 结果是一个列表列表，其中包含您得到的排序列表。

a = [
     [1,33],
     [2,10,42],
     [5,1,33,44],
     [10,42,98],
     [44,12,100,124]
    ]

res = []
while a:
    el = a.pop(0)
    res.append(el)
    for sublist in a:
        if set(el).intersection(set(sublist)):
            res[-1].extend(sublist)
            a.remove(sublist)
res = [sorted(set(i)) for i in res]
print(res)
# [[1, 5, 12, 33, 44, 100, 124], [2, 10, 42, 98]]

工作原理：

• 形成一个空的结果列表res 。 从分组a将被“转移”在这里。
• .pop()关闭的第一个元素a 。 这将修改a就位并将el定义为该元素。
• 然后遍历每个子表a ，你的突然出现比较el那些子列表和“建立”公共集。 这是您的问题有点棘手的地方，因为您需要逐渐增加相交的集合，而不是一次找到多个子列表的交集。
• 重复此过程，直到a为空。

另外，如果您只想将偶数和奇数子列表组合在一起（您的问题仍然不清楚），则可以使用itertools：

from itertools import chain
grp1 = sorted(set(chain.from_iterable(a[::2])))
grp2 = sorted(set(chain.from_iterable(a[1::2])))
print(grp1)
print(grp2)
# [1, 5, 12, 33, 44, 100, 124]
# [2, 10, 42, 98]