请采用Scala语言写出one-hot编码的案例

下面是使用Scala实现的一个简单的one-hot编码案例：

object OneHotEncoder {
  def encode(labels: List[String]): Map[String, List[Int]] = {
    val distinctLabels = labels.distinct
    val encodingMap = distinctLabels.zipWithIndex.toMap

    labels.map(label => {
      val encoding = Array.fill(distinctLabels.length)(0)
      encoding(encodingMap(label)) = 1
      (label, encoding.toList)
    }).toMap
  }
}

object Main {
  def main(args: Array[String]): Unit = {
    val labels = List("red", "blue", "green", "red", "yellow", "blue", "green")
    val encodedLabels = OneHotEncoder.encode(labels)

    encodedLabels.foreach { case (label, encoding) =>
      println(s"$label: ${encoding.mkString(", ")}")
    }
  }
}

在这个案例中，`OneHotEncoder` 对象有一个 `encode` 方法，它接受一个包含标签的列表，并返回一个映射，其中每个标签都与相应的one-hot编码表示形式对应。

在 `encode` 方法中，我们首先获取唯一的标签列表。然后，我们使用 `zipWithIndex` 方法为每个不同的标签分配一个唯一的索引，并将其存储在 `encodingMap` 中。

接下来，我们使用 `map` 方法遍历原始标签列表，并为每个标签创建一个one-hot编码表示。我们使用 `Array.fill` 方法创建一个全为0的数组，并将相应标签的索引位置设置为1，然后将其转换为列表。

最后，我们将标签和相应的编码作为键值对存储在一个新的映射中，并将其返回。

在 `Main` 对象中，我们使用一个示例的标签列表，然后调用 `OneHotEncoder.encode` 方法进行编码，并将结果打印到控制台。

希望这个案例能帮助到你！