对象列表中的乘法总和

Question

我是 python 的新手，我有一个很简单的问题。如何使用 sum() 在一行中执行以下代码

total= 0.0
for record in myList:
    a= record.a
    b= record.b
   total+= (a*b)

和 myList 对象列表，每个对象包含两个属性 a 和 b

我知道我们可以做到以下几点

sum([3,4,5]) == 3 + 4 + 5 == 12

但是如何求乘积呢？

Answer 1

你可以这样做：

total = sum(rec.a * rec.b for rec in myList)

注意缺少的 [...] 括号，通过推导避免了内存中列表的虚假构造，而是将生成器表达式传递给 sum.

你也可以把功能实用程序的厨房水槽扔在这里:)

from operator import mul, attrgetter
from itertools import starmap

total = sum(starmap(mul, map(attrgetter("a", "b"), myList)))

sum of multiplication in a list of objects