python—对象列表中乘法的总和

kknvjkwl 于 2021-09-13 发布在 Java

关注(0)|答案(1)|浏览(272)

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

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

1条答案

你可以做：

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

注意缺失的部分 [...] 括号，通过理解避免内存中列表的虚假构造，而是将生成器表达式传递给 sum .
您还可以将厨房Flume中的功能性utils扔到以下位置：）

from operator import mul, attrgetter
from itertools import starmap

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