Как правильно выполнять арифметику над экземплярами классов? Например. Суммируйте атрибут n_people для списка экземпляров класса House
Как правильно выполнять арифметику над экземплярами классов? В приведенном ниже примере класс представляет House и содержит атрибут n_people, представляющий количество людей в этом доме.
Цель состоит в том, чтобы выполнить суммирование экземпляров класса House и вернуть общее количество людей в домах. Кажется, это работает, когда экземпляры класса House складываются вместе:
class House:
def __init__(self, n_people):
self.n_people = n_people
def __add__(self, n_people):
return n_people + self.n_people
House(3) + House(4)
И терпит неудачу, когда суммируется массив экземпляров класса House:
houses = [House(i) for i in range(10)]
sum(houses)
В результате ошибка:
TypeError Traceback (most recent call last)
<ipython-input-66-197f26932b89> in <module>
1 houses = [House(i) for i in range(10)]
----> 2 sum(houses)
TypeError: unsupported operand type(s) for +: 'int' and 'House'
Как правильно выполнять арифметические действия с экземплярами классов в Python?
Ответы 1
Проблема, с которой вы столкнулись, заключается в том, что sum работает слева направо, поэтому после суммирования первых двух экземпляров House ваш код пытается выполнить int + House, когда определено только House + int.
Чтобы решить эту проблему, вам нужно определить __radd__, который вызывается в этой ситуации:
class House:
def __init__(self, n_people):
self.n_people = n_people
def __add__(self, n_people):
return n_people + self.n_people
def __radd__(self, n_people):
return n_people + self.n_people
House(3) + House(4)
houses = [House(i) for i in range(10)]
sum(houses)
Выход:
45