熊猫指数对象

Question

我希望能够在DataFrame中有一些通用对象实例的列标签。因此，而不是一个个STR对象，一个简单的类，它包装STR并提供一些额外的功能：

class WrapStr(object): 
    def __init__(self,str): 
     self.str = str 
    def __eq__(self,other): 
     return self.str == other.str 
    def __repr__(self): 
     return self.str 

的问题是，pd.Index不呼吁WrapStr实例EQ方法而只是检查是否这两个实例是相同的。

first_ins = WrapStr('col1') 
my_ix = pd.Index([first_ins]) 
sec_ins = WrapStr('col1') 

print first_ins in my_ix # True 
print sec_ins in my_ix # False 

它看起来像包含检查在https://github.com/pydata/pandas/blob/master/pandas/index.pyx线92和448

如何支持这样的扩展列标签任何想法界定？

Answer 1

添加__hash__方法WrapStr：

class WrapStr(object): 
    def __init__(self,str): 
     self.str = str 
    def __eq__(self,other): 
     return self.str == other.str 
    def __repr__(self): 
     return self.str 
    def __hash__(self): 
     return hash(self.str) 

first_ins = WrapStr('col1') 
my_ix = pd.Index([first_ins]) 
sec_ins = WrapStr('col1') 

print first_ins in my_ix # True 
print sec_ins in my_ix # False