Seq[unit] 类型的表达式不符合 scala 中的预期类型 Seq[DataFrame]

Question

在我的函数中，我返回一个finalDF，一个数据帧序列。在下面显示的循环中，map 返回Seq[DataFrame] 并且它被存储在finalDF 中以便能够返回给调用者，但是在某些需要进一步处理的情况下，我想为每次迭代存储过滤后的数据帧并将其传递给下一个循环。

我该怎么做？如果我尝试将其分配给某个临时值，它会抛出 Seq[unit] 类型的表达式不符合预期类型 Seq[DataFrame] 的错误。

var finalDF: Seq[DataFrame] =null

    for (i <- 0 until stop){
  finalDF=strataCount(i).map(x=> {
    df.filter(df(cols(i)) === x)

    //how to get the above data frame to pass on to the next computation?
    }
  )

}

问候

Answer 1

也许这会有所帮助：

val finalDF: Seq[DataFrame] = (0 to stop).flatMap(i => strataCount(i).map(x => df.filter(df(cols(i)) === x))).toSeq

flatMap 将Seq(Seq) 展平。

(0 to stop) 将从0 循环到stop，flatMap 将变平List，例如：

scala> (0 to 20).flatMap(i => List(i))
res0: scala.collection.immutable.IndexedSeq[Int] = Vector(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20)
scala> (0 to 20).map(i => List(i)).flatten
res1: scala.collection.immutable.IndexedSeq[Int] = Vector(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20)

对于两个计数器，也许你可以这样做：

(0 to stop).flatMap(j => {
 (0 to stop).flatMap(i => strataCount(i).map(x => df.filter(df(cols(i)) === x)))
}).toSeq

或尝试：for yield，参见：Scala for/yield syntax