我可以创建一个类似于不可变(浮点)的类吗？

Question

我有一个计算移动平均线的类,可能会有所改进.平均窗口的大小必须灵活.它目前的工作原理是设置窗口的大小,然后发送更新:

twoday = MovingAverage(2)    # twoday.value is None
twoday = twoday.update(10)   # twoday.value is None
twoday = twoday.update(20)   # twoday.value == 15
twoday = twoday.update(30)   # twoday.value == 25

我认为如果它更像这样的东西会很酷:

twoday = MovingAverage(2)    # twoday is None
twoday += 10                 # twoday is None
twoday += 20                 # twoday == 15
twoday += 30                 # twoday == 25

这是愚蠢的吗？可能吗？

Answer 1

您可以通过添加完全按照您想要的方式来模拟数字类型__add__().

只需添加方法,如

def __iadd__(self, other):
    self.update(other)
    return self

def __add__(self, other):
    return self.value + other

def __str__(self):
    return str(self.value)

你现在拥有的.

如果你想接近浮点数的作用,你可以添加诸如的方法

def __float__(self):
    return self.value

def __radd__(self, other):
    return other + self.value

(后者为您提供了一种方法,somevalue + twoday并获得预期的价值)

和__mul__/__rmul__div一样,等等.您唯一的特殊情况可能是__iadd__()上面提到的.