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

Question

我有 data = [[t1, t2, ...], [v1, v2, ...]] 这样的数据。我想将其包装在 class 中，这样我就可以调用 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'

如何使我的 class 与参数解包运算符一起正常工作？

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 指出这称为 "sequence protocol"。

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

How to overload * argument unpacking operator?

python

arguments

argument-unpacking