Spark Scala MLlib 赋值语法

Question

我一直在阅读https://spark.apache.org/docs/latest/ml-statistics.html 的指南，我注意到他们正在使用这种语法进行 val 赋值：

val Row(coeff1: Matrix) = Correlation.corr(df, "features").head

谁能详细说明这意味着什么？它似乎类似于 Scala 处理正则表达式组提取的方式......

Answer 1

它只不过是一个pattern matching。为了使其更明显，您将其重写为：

val coeff1 = Correlation.corr(df, "features").head match {
  case Row(coeff1: Matrix) => coeff1
}

换句话说，它只是尝试匹配从.head 调用返回的对象，并且在成功匹配时，它会创建一个对包含在返回的Row 中的Matrix 对象的引用（coeff1）。