gpt4 book ai didi

python - Python 中内置类型的自定义比较函数

转载 作者:太空狗 更新时间:2023-10-30 02:04:55 26 4
gpt4 key购买 nike

我正在使用 Python 的内置集合来保存我定义的类的对象。对于此类,我定义了 __eq____ne____hash__,以便我可以通过自定义比较函数来比较对象。这工作得很好,直到我发现我实际上需要两组比较函数,它们将在我的代码中的不同时间以不同的方式使用。

我无法在我的类中定义两组__eq__ 等方法,并且Python 的内置集合类型不接受比较器参数。我想我可以围绕 set 编写一个包装器类,但这似乎比必要的工作要多得多。

有没有比编写我自己的 set 类更简单的解决方案?

最佳答案

假设你有这个类:

class Thingy(object):
def __init__(self, key, notkey):
self.key, self.notkey = key, notkey
def __eq__(self, other):
return self.key == other.key
def __hash__(self):
return hash(self.key)

现在,您想将它们放在一个集合中,但使用 notkey 而不是 key 作为键。你不能按原样这样做,因为一个集合期望它的元素具有一致的相等含义 - 以及哈希的一致含义这样 a == b 总是暗示 hash (a) == 散列(b)。因此,创建一个包装器:

class WrappedThingy(object):
def __init__(self, thingy):
self.thingy = thingy
def __eq__(self, other):
return self.thingy.notkey == other.thingy.notkey
def __hash__(self):
return hash(self.thingy.notkey)

您可以将那些放在一个集合中:

wts = set(WrappedThingy(thingy) for thingy in thingies)

例如,假设您想要唯一化您的事物,为每个 notkey 值保留一个事物(任意)。只需将它们包裹起来,将 wrapper 粘在一起,然后打开它们并将展开的 wrapper 粘在一个列表中:

wts = set(WrappedThingy(thingy) for thingy in thingies)
thingies = [wt.thingy for wt in wts]

这是称为“DSU”的更通用的 Python 模式的一部分。这代表“decorate-sort-undecorate”,现在非常不准确,因为在现代 Python 中你几乎不需要它来完成与排序相关的任务……但从历史上看它是有意义的。随意称它为“decorate-process-undecorate”,希望它能流行起来,但不要抱太大希望。

现在排序不需要 DSU 的原因是大多数排序函数都将 key 函数作为参数。事实上,即使是统一化,itertools recipes 中的 unique_everseen 函数也是如此接受一个key

但是如果你看一下它在幕后做了什么,它基本上就是 DSU:

for element in iterable:
k = key(element)
if k not in seen:
seen.add(k)
yield element

(事实上,它是一个生成器而不是一个列表构建函数,这意味着它可以“即时取消修饰”,这让事情变得更简单。但除此之外,想法相同。)

关于python - Python 中内置类型的自定义比较函数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/16306844/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com