在Rails中如何对多个对象的两个值乘积求和?
如何用Ruby简洁计算投资组合中PortStock的成交量与买入价乘积总和
嘿,这个问题其实很好解决,而且Ruby正好有非常简洁的方式来实现——甚至比你用map(&:total_spend).sum的思路更直接!
最简洁高效的写法
Ruby的sum方法本身就支持传入一个块,直接对每个元素执行计算后累加,完全不需要先map再sum(还能避免生成中间数组,处理大量数据时性能更好):
p.port_stocks.sum { |ps| ps.volume * ps.purchase_price }
用你的示例数据测试的话,这个代码会自动遍历每一个PortStock对象:
- 第一个对象计算
2000 * 5.9 = 11800 - 第二个对象计算
1000 * 9.0 = 9000
最后直接返回两者的总和19800,和你手动计算的结果完全一致。
如果你更习惯map+sum的写法
如果你还是偏好先转换再求和的思路,也可以用你熟悉的模式,把块替换成自定义的计算逻辑:
p.port_stocks.map { |ps| ps.volume * ps.purchase_price }.sum
不过要注意,这种方式会先创建一个包含所有乘积结果的数组,再对数组求和。当处理500个甚至更多对象时,前面的sum带块的方式会更高效。
进阶优化:给PortStock添加自定义方法
如果这个计算逻辑会频繁用到,你可以给PortStock类加一个专属方法,让代码更具可读性:
class PortStock def volume_price_product volume * purchase_price end end
之后就可以用和total_spend完全一致的风格来求和了:
p.port_stocks.sum(&:volume_price_product)
不管你的投资组合里有2个还是500个PortStock对象,以上几种写法都能完美适配,完全符合Ruby简洁优雅的风格。
内容的提问来源于stack exchange,提问作者marcamillion
相关产品推荐
相关产品推荐

