Как суммировать атрибуты всех экземпляров объекта

Question

Я хочу суммировать атрибут costum для всех экземпляров объекта.

class ActivityCenter:

    def __init__(self, costpool, costsum, costdriver, cdunits):
        self.costpool = costpool
        self.costsum = costsum
        self.costdriver = costdriver
        self.cdunits = cdunits

cp1 = ActivityCenter("Material Handling", 480000, "Pounds", 160000)
cp2 = ActivityCenter("Production orders", 90000, "Num of POs", 200)

# I am looking to add up the costsum values for all instances, similar to:
costsumtotal = (cp1.__dict__.get("costsum")) + (cp2.__dict__.get("costsum"))

До сих пор я пытался использовать sum() с пониманием следующим образом, обращаясь к этому решению:

B = []
for i in range(10):
    B.append(ActivityCenter())

s = sum(i.costsum for i in B)

Но у меня возникают проблемы с преодолением ошибки TypeError, из-за которой я пропускаю 4 обязательных позиционных аргумента.

Answer 1

Утилизировать sum Встроенная функция в Python для переменных-членов объектов, вам необходимо составить последовательность (например, кортеж или список) переменных-членов объектов. В следующем фрагменте показано, как составить список переменных-членов объектов. Код, который вы разместили, опускает выражение понимания. Я надеюсь, что это будет полезно:)

class ActivityCenter:

    def __init__(self, costpool, costsum, costdriver, cdunits):
        self.costpool = costpool
        self.costsum = costsum
        self.costdriver = costdriver
        self.cdunits = cdunits

"""
Create some objects

objs = []
for i in range(num_obj):
    objs.append(ActivityCenter(<some value>,<...>,<...>,<...>))

Or use objects to make a list
"""
cp1 = ActivityCenter("Material Handling", 480000, 160000, "Pounds")
cp2 = ActivityCenter("Production orders", 90000, 200, "Num of POs")
cp3 = ActivityCenter("Marketing", 120000, 1000, "Num of events")

objs = [cp1, cp2, cp3]

total_cost = sum([obj.costsum for obj in objs])  # List comprehension
print("Total cost: ", total_cost)