Python字典键(它是类对象)与多个比较器进行比较

时间:2012-03-23 07:48:11

标签: python hash dictionary

我在python字典中使用自定义对象作为键。这些对象定义了一些默认的 hash eq 方法,这些方法在默认比较中使用 但在某些功能中,我需要使用不同的方法来比较这些对象。 那么有没有办法为这个特定的函数覆盖或传递一个新的比较器来进行这些键比较。

更新:我的班级有以下类型的功能(这里我无法编辑哈希方法,它会在其他地方影响很多)

class test(object):

    def __init__(self,name,city):
        self.name=name
        self.city=city

    def __eq__(self,other):
        hash_equality= (self.name==other.name)
        if(not hash_equality):
            #check with lower
            return (self.name.lower()==other.name.lower())


    def  __hash__(self):
        return self.name.__hash__()

my_dict={}
a=test("a","city1")
my_dict[a]="obj1"
b=test("a","city2")
print b in my_dict  #prints true
c=test("A","city1")
print c in my_dict  #prints false
print c in my_dict.keys() #prints true
# my_dict[c]   throw error

这是正常的功能。但在一个特定的方法中,我想覆盖/或传递一个新的自定义比较器,其中新的哈希代码就像

def  __hash__(self):
    return self.name.lower().__hash__()

以便c in my_dict返回ture

my_dict[c] will return "obj1"

很抱歉这么多更新。

在排序中我们可以将自定义方法作为比较器传递,有没有办法在这里做同样的事情。

5 个答案:

答案 0 :(得分:4)

使这项工作的唯一方法是使用新的哈希和比较函数创建字典的副本。原因是字典需要使用新的哈希函数重新散列每个存储的密钥,以使查找按照您的意愿工作。由于您无法为字典提供自定义哈希函数(它总是使用其中一个键对象),因此最好的办法是将对象包装在使用自定义哈希和比较函数的类型中。

class WrapKey(object):
    __init__(self, wrapee):
        self._wrapee = wrapee

    __hash__(self):
        return self._wrapee.name.lower().__hash__()

    __eq__(self, other):
        return self._wrapee.name == other._wrapee.name


def func(d):
    d_copy = dict((WrapKey(key), value) for key, value in d.iteritems())
    # d_copy will now ignore case

答案 1 :(得分:0)

查看您可以在对象中定义的comparison methods

根据您的想法,__cmp__可能也很有趣。

答案 2 :(得分:0)

对于这种情况有点破解:

class test(object):

    def __init__(self,name,city,hash_func=None):
        self.name=name
        self.city=city
        self.hash_func = hash_func

    def __eq__(self,other):
        return self.__hash__()==other.__hash__()

    def  __hash__(self):
        if self.hash_func is None:
            return self.name.__hash__()
        else:
            return self.hash_func(self)

my_dict={}
a=test("a","city1")
my_dict[a]="obj1"
b=test("a","city2")
print b in my_dict  #prints true
c=test("A","city1")
print c in my_dict  #Prints false
c.hash_func = lambda x: x.name.lower().__hash__()
print c in my_dict #Now it prints true

您无法更改存储在dict中的哈希值,但您可以更改哈希用于查找。当然,这会导致像这样的奇怪的东西

my_dict={}
a=test("a","city1")
my_dict[a]="obj1"
a.hash_func = lambda x: 1
for key in my_dict:
    print key in my_dict # False

答案 3 :(得分:0)

现在我正在使用自定义dict(派生类的dict),它将comparer作为参数,我已经覆盖了 contains getitems (),它们会检查并给出基于值的值在比较器上。

答案 4 :(得分:0)

步骤:实现自定义键类并覆盖哈希和相等函数。

e.g。

class CustomDictKey(object):

def __init__(self, 
             param1,
             param2):

            self._param1 = param1
            self._param2 = param2

 # overriding hash and equality function does the trick

def __hash__(self):
    return hash((self._param1,
             self._param2))

def __eq__(self, other):
    return ( ( self._param1,
             self._param2 ) == ( other._param1,
             other._param2) )

def __str__(self):
    return "param 1: {0} param 2: {1}  ".format(self._param1, self._param2)

主要方法

if name == 'main':

    # create custom key
    k1  = CustomDictKey(10,5)

    k2  = CustomDictKey (2, 4)

    dictionary = {}

    #insert elements in dictionary with custom key
    dictionary[k1] = 10
    dictionary[k2] = 20

    # access dictionary values with custom keys and print values
    print "key: ", k1, "val :", dictionary[k1]
    print "key: ", k2, "val :", dictionary[k2]

有关详细信息,请参阅链接 Using custom class as key in Python dictionary