Могу ли я изменить значение базового класса в подклассе Python? - PullRequest
Новая
       2

Могу ли я изменить значение базового класса в подклассе Python?

1 голос
/ 07 июля 2011

При создании подкласса базового типа, такого как float, возможно ли «пересчитать» или «переназначить» исходное значение?Если у меня есть следующее определение класса, 

import collections

class MovingAverage(float):
    def __new__(self, initial_value, window):
        self.d = collections.deque([initial_value], window)
        return float.__new__(self, initial_value)
    def add(self, new_value):
        self.d.append(new_value)
        print sum(self.d) / len(self.d)
        # here, I want to _set_ the value of MovingAverage to
        #     sum(self.d) / len(self.d)

Когда я начинаю с 

>>> ma = MovingAverage(10, 3)
>>> ma
10.0

, но 

>>> ma.add(3)
6.5
>>> ma
10.0

Другое определение класса, которое я пробовал, это: 

import collections

class MovingAverage(float):
    def __new__(self, iterable, window):
        self.d = collections.deque(iterable, window)
        initial_value = sum(iterable) / len(iterable)
        return float.__new__(self, initial_value)
    def add(self, new_value):
        self.d.append(new_value)
        return MovingAverage(self.d, self.d.maxlen)

На этот раз, когда я начну с 

>>> ma = MovingAverage([10], 3)
>>> ma
10.0

и 

>>> ma.add(3)
6.5
>>> ma
10.0
>>> ma = ma.add(3)
>>> ma
5.333333333333333

Однако, я думаю (я не проверял, чтобы выяснить это)это делает это значительно медленнее.Так можно ли это сделать?Можно ли как-то установить его так, чтобы возвращаемое значение ma было искомым значением?Или мне нужно определить атрибут value, изменить базовый класс на object и отказаться от своего предлога о том, что у меня есть шанс контролировать возвращаемое значение класса?

Ответы [ 2 ]

5 голосов
/ 07 июля 2011

Нет.Поскольку эти типы неизменяемы, вы должны использовать encapsulation , а не наследование.

2 голосов
/ 07 июля 2011

Не думаю, что ответ Игнасиоса, приведенный выше, нуждается в каких-либо улучшениях, но, поскольку у меня был этот класс, я просто подумал, что должен поделиться.Это позволяет избежать многократного выполнения операций с большими суммами, а также позволяет избежать ошибок округления, которые могут возникнуть при использовании (более) наивного алгоритма: 

class MovingAverage:
    def __init__(self):
        self.sum = None
        self.num = 0

    def add(self, val):
        if self.sum is None:
            self.sum = val
        else:
            self.sum += val
        self.num += 1
        return self.val

    @property
    def val(self):
        if self.sum is None:
            return None
        else:
            return self.sum/self.num

if __name__ == "__main__":
    print("Running small test suite")
    fail = False

    m = MovingAverage()
    try:
        assert m.val is None, "A new MovingAverage should be None"
        assert m.add(10) == 10, "New average should be returned"
        assert m.val == 10, "The average should update for each add"
        assert m.add(20) == 15, "New average should be returned"
        assert m.val == 15, "The average should update for each add"
        assert m.add(0) == 10, "Adding zero should work"
        assert m.add(-10) == 5, "Adding a negative number should work"
        assert m.add(-1) == 19/5, "Result should be able to be a fraction"
    except AssertionError as e:
        print("Fail: %s" % e.args[0])
        fail = True

    if not fail: print("Pass")
...