scala - 在 2 个 for 循环中使用 map 和 zip 使其更具功能性

Question

我实现了以下代码来使用 for 循环计算加权平均值，我怎样才能更 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)

输出将是 Seq(3.4, 4.4, 5.4) 因为(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)