来自对象的 Pandas 索引

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

看起来 contains 检查是在https://github.com/pydata/pandas/blob/master/pandas/index.pyx 第 92 和 448 行中定义的。

关于如何支持此类扩展列标签的任何想法？

Answer 1

给WrapStr添加一个__hash__方法：

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