我在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"
很抱歉这么多更新。
在排序中我们可以将自定义方法作为比较器传递,有没有办法在这里做同样的事情。
答案 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 。