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

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其中明确指出

Note: for loops expect that an IndexError will be raised for illegal indexes to allow proper detection of the end of the sequence.

https://docs.python.org/2/library/functions.html#iter指出这被称为“序列协议(protocol)”。