python - 在列表/集合列表中查找、收集重复项

Question

在 Python 中，我有一个元组列表和一个长度相同的整数列表，例如，

a = [
    [1, 2],
    [3, 2],
    [4, 66],
    [2, 3]
    ]

b = [
    1,
    31,
    31,
    44
    ]

a 中的第 k 个条目可以认为是与 b 中的第 k 个条目相关联。

条目 [3, 2] 和 [2, 3] 对我来说真的是一样的，我想要 a考虑到这一点。另外，我想要一个属于新的唯一列表的条目列表。对于上面的例子，

a2 = [
    [1, 2],
    [3, 2],  # or [2, 3]
    [4, 66]
    ]

b2 = [
    [1],
    [31, 44],
    [31]
    ]

b2[0] 是 [1]，因为 [1, 2] 仅与 1 关联. b2[1] 是 [31, 44] 因为 [2, 3] (等于 [3, 2] 与 a 中的 31 和 44 相关联。

可以一个接一个地遍历一个条目，使每个2-list成为一个frozenset，将其分类到一个字典中等等。不用说，如果，这不会很好地执行a 和 b 很大。

关于如何更聪明地处理这个问题的任何提示？ (列出理解？)

Answer 1

如果你想维持秩序和分组，我认为你不会比使用 OrderedDict 分组更好:

from collections  import OrderedDict
a = [
    [1, 2],
    [3, 2],
    [4, 66],
    [2, 3]
    ]

b = [1, 31, 31, 44]
d = OrderedDict()
for ind, f in enumerate(map(frozenset, a)):
        d.setdefault(f, []).append(b[ind])

print(list(d), list(d.values()))

这会给你:

[frozenset({1, 2}), frozenset({2, 3}), frozenset({66, 4})] [[1], [31, 44], [31]]

如果看到的顺序无关紧要，请使用defaultdict:

from collections  import defaultdict
a = [
    [1, 2],
    [3, 2],
    [4, 66],
    [2, 3]
    ]

b = [1, 31, 31, 44]
d = defaultdict(list)
for ind, f in enumerate(map(frozenset, a)):
        d[f].append(b[ind])

print(list(d), list(d.values()))

这会给你:

 [frozenset({1, 2}), frozenset({2, 3}), frozenset({66, 4})] [[1], [31, 44], [31]]

如果你真的想要列表或元组:

print(list(map(list, d)), list(d.values()))

这会给你:

[[1, 2], [2, 3], [66, 4]] [[1], [31, 44], [31]]

对于 python2，您应该使用 itertools.izip 和 itertools.imap 代替 map 和 zip。