在2 for循环中，使用map和zip可以更具func风格

Question

我实现了以下代码来计算加权平均值。对于循环，我如何才能更具func编程风格并使用map和zip？

val aggAvg = (emb: Seq[Seq[Float]], weights: Seq[Float]) => {
val embSize = emb.head.size
val len = emb.size
(0 until embSize)
.map { i =>
(0 until len).map { j =>
emb(j)(i) * weights(j)
}.sum / weights.sum
}
}

示例：给定

val emb: Seq[Seq[Float]] = Seq(Seq(1,2,3), Seq(4,5,6))
val weights: Seq[Float] = Seq(2, 8)

输出将是CCD_ 3，因为(1 * 2 + 4 * 8) / (2 + 8) = 3.4等。

Answer 1

这里有一种方法，尽管我不确定它是否是最优雅的

val aggAvg = (emb: Seq[Seq[Float]], weights: Seq[Float]) =>
emb.transpose.map((weights, _).zipped.map(_ * _).sum).map(_ / weights.sum)
res0: Seq[Float] = List(3.4, 4.4, 5.4)

Answer 2

这使用Iterator代替zip操作。

val aggAvg = (emb: Seq[Seq[Float]], weights: Seq[Float]) => {
val wtItr = weights.iterator
val wtSum = weights.sum
emb.map(_.map(wtItr.next().*)).transpose.map(_.sum/wtSum)
}

注意：如果不是所有的行长度都与weights长度匹配，这可能会有点危险，但您发布的aggAvg代码在此规则中是相同的，所以我省略了安全检查。