问题
我想知道是否有任何 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}
