Question

是否可以使用初始内存引用重载[] (__getitem__) Python运算符和链方法。

想象一下，我有一个类Math接受一个整数列表，如下所示：

class Math(object):
    def __init__(self, *args, **kwargs):
        assert(all([isinstance(item, int) for item in list(args)]))
        self.list = list(args)

    def add_one(self):
        for index in range(len(self.list)):
            self.list[index] += 1

我想做这样的事情：

instance = Math(1,2,3,4,5)
instance[2:4].add_one()

执行此代码后instance.list应为[1,2,4,5,5]，这可能吗？

我知道我可以做add_one(2,4)之类的事情，但如果可能的话，这不是我希望拥有的API风格。

由于

Answer 1

正如温斯顿所提到的，你需要实现一个辅助对象：

class Math(object):
    def __init__(self, *args, **kwargs):
        self.list = list(args)

    def __getitem__(self, i):
        return MathSlice(self, i)

class MathSlice(object):
    def __init__(self, math, slice):
        self.math = math
        self.slice = slice

    def add_one(self):
        for i in xrange(*self.slice.indices(len(self.math.list))):
            self.math.list[i] += 1


instance = Math(1,2,3,4,5)
instance[2:4].add_one()

print instance.list

如何与MathSlice对象共享数学对象取决于数学对象更改时您希望语义是什么。

Answer 2

Numpy做了这样的事情。

__getitem__方法将收到slice个对象。有关详细信息，请参阅http://docs.python.org/reference/datamodel.html。您需要返回一个新对象，但要实现该对象以便修改原始列表。

使用内存引用重载[] python运算符和链接方法

2 个答案: