如何在Spark中对具有日期和时间值的列进行排序?

时间:2016-11-17 12:49:08

标签: apache-spark dataframe apache-spark-sql rdd

注意: 我把它作为Spark中的Dataframe。 此时间/日期值构成Dataframe中的单个列。

输入:

  

04-NOV-16 03.36.13.000000000 PM
  06-NOV-15 03.42.21.000000000 PM
  05-NOV-15 03.32.05.000000000 PM
  06-NOV-15 03.32.14.000000000 AM

预期产出:

05-NOV-15 03.32.05.000000000 PM
06-NOV-15 03.32.14.000000000 AM
06-NOV-15 03.42.21.000000000 PM
04-NOV-16 03.36.13.000000000 PM

2 个答案:

答案 0 :(得分:12)

由于此格式不是标准格式,您需要使用unix_timestamp函数来解析字符串并转换为时间戳类型:

import org.apache.spark.sql.functions._

// Example data
val df = Seq(
  Tuple1("04-NOV-16 03.36.13.000000000 PM"),
  Tuple1("06-NOV-15 03.42.21.000000000 PM"),
  Tuple1("05-NOV-15 03.32.05.000000000 PM"),
  Tuple1("06-NOV-15 03.32.14.000000000 AM")
).toDF("stringCol")

// Timestamp pattern found in string
val pattern = "dd-MMM-yy hh.mm.ss.S a"

// Creating new DataFrame and ordering
val newDF = df
  .withColumn("timestampCol", unix_timestamp(df("stringCol"), pattern).cast("timestamp"))
  .orderBy("timestampCol")

newDF.show(false)

结果:

+-------------------------------+---------------------+
|stringCol                      |timestampCol         |
+-------------------------------+---------------------+
|05-NOV-15 03.32.05.000000000 PM|2015-11-05 15:32:05.0|
|06-NOV-15 03.32.14.000000000 AM|2015-11-06 03:32:14.0|
|06-NOV-15 03.42.21.000000000 PM|2015-11-06 15:42:21.0|
|04-NOV-16 03.36.13.000000000 PM|2016-11-04 15:36:13.0|
+-------------------------------+---------------------+

有关unix_timestamp和其他实用程序函数的更多信息,请参见here

要构建时间戳格式,可以参考SimpleDateFormatter docs

正如pheeleeppoo所说的

编辑1:,您可以直接通过表达式进行排序,而不是创建新列,假设您只想在数据帧中保留字符串类型的列:

val newDF = df.orderBy(unix_timestamp(df("stringCol"), pattern).cast("timestamp"))

编辑2:请注意,unix_timestamp函数的精度以秒为单位,因此如果毫秒非常重要,则可以使用udf:

def myUDF(p: String) = udf(
  (value: String) => {
    val dateFormat = new SimpleDateFormat(p)
    val parsedDate = dateFormat.parse(value)
    new java.sql.Timestamp(parsedDate.getTime())
  }
)

val pattern = "dd-MMM-yy hh.mm.ss.S a"
val newDF = df.withColumn("timestampCol", myUDF(pattern)(df("stringCol"))).orderBy("timestampCol")

答案 1 :(得分:2)

字符串转换为时间戳后,您还可以使用排序功能:

   df.sort(unix_timestamp(df("dateColumn"), "dd-MMM-yy hh.mm.ss.S a").cast("timestamp"))
     .show(false)