Skip to main content
 首页 » 编程设计

python之是否有任何带有 key 参数的 set() Python 实现

2025年05月04日18yjmyzz

问题

我想知道是否有任何 python 函数使用 key 参数实现 set(),类似于例如sorted(iterable, *, key=None, reverse=False) 中的 key?

我认为此功能在某些情况下会很有用,尤其是在无需更改其 hash() 函数的情况下比较自定义对象时。

期望的效果

示例 1

iterable = [1.0, 1.5, 2.0, 2.5] 
x = set(iterable)                 # yields {1.0, 1.5, 2.0, 2.5} 
y = set(iterable, key=math.floor) # yields {1.0, 2.0} 

示例 2

iterable = ['abc', 'ade', 'bbc', 'abc', 'bfg'] 
x = set(iterable)                     # yields {'abc', 'ade', 'bbc', 'bfg'} 
y = set(iterable, key=lambda x: x[0]) # yields {'abc', 'bbc'} 

选择哪个

我可以想象这不会实现,因为如果找到重复项,则很难选择选择哪个对象。然而,这可以通过一些策略来克服,例如'先选择', '随机选择', '最后选择'。

回答

感谢您的所有回复,Daniel Roseman 提出了针对此问题的最优雅的解决方案:

key = math.floor # Or any desired key function 
y   = set({key(i): i for i in iterable}.values()) 

请您参考如下方法:

这不就是一本字典吗,其中的键是将函数应用于值的结果?

y = {math.floor(i): i for i in iterable}