Question

我有像data = [[t1, t2, ...], [v1, v2, ...]]这样的数据。我想将它包装在一个类中，这样我就可以调用data.t而不必使用data[0]。

我尝试使用以下方法执行此操作：

class Variable:
    def __init__(self, data):
        self.t = data[0]
        self.v = data[1]

    def __getitem__(self, key):
        if key == 0:
            return self.t
        elif key == 1:
            return self.v
        else:
            raise ValueError("not valid key '{}'".format(key))

    def __setitem__(self, key, value):
        if key == 0:
            self.t = value
        elif key == 1:
            self.v = value
        else:
            raise ValueError("not valid key '{}'".format(key))

__getitem__和__setitem__重载的原因是为了向后兼容，因此data[0]仍然有效。这适用于大多数事情，但是我遇到了以下调用的问题：

func_that_takes_two_arguments(*data) # unpacking data

我得到的错误是

/Users/pingul/Workspace/lhcfill/oml.py in __getitem__(self, key)
     52                                 return self.val
     53                         else:
---> 54                                 raise ValueError("not valid key     '{}'".format(key))
     55 
     56                 def __setitem__(self, key, value):
ValueError: not valid key '2'

如何使用参数解包操作符使我的类正常工作？

Answer 1

*运算符通过迭代对象来工作。只需实现__getitem__()即可完成此迭代，但您的实现有问题。相反，如果提出ValueError，则应抛出IndexError，表示迭代结束。

另请参阅明确说明的https://docs.python.org/3/reference/datamodel.html#object.getitem

注意： for循环期望为非法索引引发IndexError以允许正确检测序列的结尾。

https://docs.python.org/2/library/functions.html#iter表示这被称为＆＃34;序列协议＆＃34;。

如何重载*参数解包运算符？

1 个答案: