通过特定元素创建一系列元组

时间:2015-07-19 08:20:03

标签: python set tuples

所以我有一个元组元组

a = ((1, 2), (7, 2), (5, 2), (3, 4), (8, 4))

我想从'a'中删除所有具有共同第二个元素的元组,除了一个(其中任何一个)。

对于上面的例子,我想要新输出a = ((1,2),(3,4))

换句话说,我想消除在元组的第二个位置被认为是重复元素的元组。

我想知道实现这一目标的最有效方法,并且想知道我是否可以使用列表而不是元组执行相同操作?

1 个答案:

答案 0 :(得分:6)

您可以从元素中创建词典,包括您想要作为键的唯一内容,然后提取值。这适用于'unique'子元素可以清洗的任何东西。整数是可以清除的:

def unique_by_key(elements, key=None):
    if key is None:
        # no key: the whole element must be unique
        key = lambda e: e
    return {key(el): el for el in elements}.values()

这个功能非常通用;它可以用于通过任何特征提取“唯一”元素,只要key可调用返回可以用作字典中的键。订单不会被保留,目前每个密钥的最后一个元素都会获胜。

使用上述功能,您可以使用operator.itemgetter() object或lambda从每个元素中提取第二个值。这适用于一系列元组和一系列列表:

from operator import itemgetter

unique_by_second_element = unique_by_key(a, key=itemgetter(1))

演示:

>>> from operator import itemgetter
>>> a = ((1, 2), (7, 2), (5, 2), (3, 4), (8, 4))
>>> unique_by_key(a, key=itemgetter(1))
[(5, 2), (8, 4)]
>>> b = [[1, 2], [7, 2], [5, 2], [3, 4], [8, 4]]
>>> unique_by_key(b, key=itemgetter(1))
[[5, 2], [8, 4]]

请注意,该函数始终返回列表;你总是可以通过在结果上调用tuple()来转换回来。

相关问题