algorithm - Scala:如何根据添加到第三个列表的两个列表的加权差异创建一个新列表

Question

作为 Differential Evolution 实现的一部分我需要实现“突变”步骤的算法:

1. 从群体中随机挑选三个成员，他们必须彼此不同，并且与给定成员不同
2. 通过将两个向量的加权差与第三个向量相加来计算施主成员

This是我想出的:

// Algorithm types
type Member = List[Double]
type Generation = Vector[Member]

def mutate(index: Int, generation: Generation): Member = {
  // Create a random number stream with distinct values
  val selector = Stream.continually(Random.nextInt(N)).distinct

  // Select 3 mates from the generation
  val mates = selector.filter(_ != index).take(3).map(generation(_))

  // Calculate the donor member
  (mates(0), mates(1), mates(2)).zipped map {
    case (e1, e2, e3) => e1 + F * (e2 - e3)
  }
}

(我按照说明实现了算法 here )

现在我的问题；有没有更好的方法来实现这一步？我一直在尝试找到一种更好的方法来从向量中选择 3 个列表并将它们压缩在一起，但是除了手动将所选列表放入元组之外我找不到任何其他方法。 Scala 编译器发出警告，应该使用 mates.head 而不是 mates(0)，这让我知道这可以用更优雅的方式实现。

提前致谢!

Answer 1

您可以转置您的伙伴，然后使用Seq 提取器映射:

mates.transpose map {
  case Seq(e1, e2, e3) => e1 + F * (e2 - e3)
}

这将是一个 Stream[Double]，因此要获得一个 Member，您必须对其调用 toList，或者使用 mates.toList.transpose ...