保持冻结集中元素的顺序

Question

我有一个元组列表，每个元组包含一个字符串和两个整数。该列表如下所示：

x = [('a',1,2), ('b',3,4), ('x',5,6), ('a',2,1)]

该列表包含数千个这样的元组。现在，如果我想获得独特的组合，我可以frozenset按如下方式在我的列表中进行：

y = set(map(frozenset, x))

这给了我以下结果：

{frozenset({'a', 2, 1}), frozenset({'x', 5, 6}), frozenset({3, 'b', 4})}

我知道 set 是一个无序的数据结构，这是正常情况，但我想在此处保留元素的顺序，以便此后可以将元素插入到pandas数据帧中。数据框将如下所示：

 Name  Marks1  Marks2
0    a       1       2
1    b       3       4
2    x       5       6

Answer 1

您可以将其仅用作辅助数据结构，而不是直接对setof进行操作 - 就像itertools 部分的配方中一样（逐字复制）：frozensetunique_everseen

from itertools import filterfalse

def unique_everseen(iterable, key=None):
    "List unique elements, preserving order. Remember all elements ever seen."
    # unique_everseen('AAAABBBCCDAABBB') --> A B C D
    # unique_everseen('ABBCcAD', str.lower) --> A B C D
    seen = set()
    seen_add = seen.add
    if key is None:
        for element in filterfalse(seen.__contains__, iterable):
            seen_add(element)
            yield element
    else:
        for element in iterable:
            k = key(element)
            if k not in seen:
                seen_add(k)
                yield element

基本上，当您使用以下命令时，这可以解决问题key=frozenset：

>>> x = [('a',1,2), ('b',3,4), ('x',5,6), ('a',2,1)]

>>> list(unique_everseen(x, key=frozenset))
[('a', 1, 2), ('b', 3, 4), ('x', 5, 6)]

这会按原样返回元素，并且还会维护元素之间的相对顺序。