对象列表中的乘法和

问题描述

我是 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

但是如何得到乘法的总和？

解决方法

你可以这样做：

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)))