比较 itemgetter 对象

Question

我注意到 operator.itemgetter 对象没有定义 __eq__，因此它们的比较默认为检查身份 (is)。

当两个itemgetter 实例的初始化参数列表比较相等时，将它们定义为相等有什么缺点吗？

这是这种比较的一个用例。假设您定义了一个排序的数据结构，其构造函数需要一个键函数来定义排序。假设您要检查两个这样的数据结构是否具有相同的关键功能（例如，在 assert 语句中；或验证它们是否可以安全合并；等等）。

如果我们能在两个关键函数是itemgetter('id') 时肯定地回答这个问题，那就太好了。但目前，itemgetter('id') == itemgetter('id') 将评估为 False。

Answer 1

Niklas 的回答非常聪明，但需要一个更强的条件，因为itemgetter 可以接受多个参数

from collections import defaultdict
from operator import itemgetter
from itertools import count

def cmp_getters(ig1, ig2):
   if any(not isinstance(x, itemgetter) for x in (ig1, ig2)):
      return False
   d1 = defaultdict(count().next)
   d2 = defaultdict(count().next)
   ig1(d1)                                 # populate d1 as a sideeffect
   ig2(d2)                                 # populate d2 as a sideeffect
   return d1==d2

一些测试用例

>>> cmp_getters(itemgetter('foo'), itemgetter('bar'))
False
>>> cmp_getters(itemgetter('foo'), itemgetter('bar','foo'))
False
>>> cmp_getters(itemgetter('foo','bar'), itemgetter('bar','foo'))
False
>>> cmp_getters(itemgetter('bar','foo'), itemgetter('bar','foo'))
True

Answer 2

itemgetter 返回一个可调用对象。我希望您不想比较可调用对象。正确的？因为即使您传递相同的参数，返回的可调用对象的 id 也不能保证相同。

def fun(a):
    def bar(b):
        return a*b
    return bar

a = fun(10)
print id(a(10))

a = fun(10)
print id(a(10))

另一方面，当您使用 itemgetter callable 作为访问器来访问底层对象时，该对象的比较将用于执行比较 它在Sorting Howto using the operating module functions 中进行了说明。