scala - 在 Scala 中获取两个相似列的最小值或最大值

Question

我刚开始学习 Scala，我正试图找出一种方法来获取 min 中相同类型的两个或多个 Columns 的 DataFrame 。我有以下代码，它分别为我提供了 min 的 max 和 Column。

inputDF.select(min($"dropoff_longitude")).show
inputDF.select(max($"pickup_longitude")).show

我如何获得 min 、 Columns 和 dropoff_longitude 的 pickup_longitude 。我是这样做的

scala.math.min(
   inputDF.select(min($"pickup_longitude")).head.getFloat(0),
   inputDF.select(min($"dropoff_longitude")).head.getFloat(0)
)

有一个更好的方法吗？

谢谢

Answer 1

认为你可以做这样的事情:

// assume all columns are of type Int, it could be other numeric type as well
val df = Seq((1,2),(3,4)).toDF("A", "B")    

df.agg(min($"A"), min($"B")).first().toSeq.map{ case x: Int => x }.min
// res14: Int = 1