您可以从元素中创建一个 字典 ,并使用任何您希望唯一的键作为键,然后提取值。这适用于“唯一”子元素可哈希化的任何内容。整数是可哈希的:
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()对象或lambda从每个元素中提取第二个值。然后,这对于元组序列和列表序列均适用:
from operator import itemgetterunique_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()结果将其转换回去。



