2016-11-29 306 views
0

我有一个数据帧在unixtime中有两个列日期,我想查找这两列之间的星期差异。 SparkSQL中有一个weekOfYear UDF,但只有在两个日期都在同一年的时候才有用。那我怎样才能找到周差呢?如何找到两个日期之间的星期差异

p.s.我正在使用Scala Spark。

+0

Spark的版本是什么? –

+0

@HZ:考虑接受最好的答案。 – mrsrinivas

回答

1

您可以采取的办法为这个创建一个自定义的UDF的:

scala> val df=sc.parallelize(Seq((1480401142453L,1480399932853L))).toDF("date1","date2") 
df: org.apache.spark.sql.DataFrame = [date1: bigint, date2: bigint] 

scala> df.show 
+-------------+-------------+ 
|  date1|  date2| 
+-------------+-------------+ 
|1480401142453|1480399932853| 
+-------------+-------------+ 


scala> val udfDateDifference=udf((date1:Long,date2:Long)=>((date1-date2)/(60*60*24*7)).toInt 
    | 
    |) 
udfDateDifference: org.apache.spark.sql.expressions.UserDefinedFunction = UserDefinedFunction(<function2>,IntegerType,Some(List(LongType, LongType))) 

scala> val resultDF=df.withColumn("dateDiffernece",udfDateDifference(df("date1"),df("date2"))) 
resultDF: org.apache.spark.sql.DataFrame = [date1: bigint, date2: bigint ... 1 more field] 

scala> resultDF.show 
+-------------+-------------+--------------+ 
|  date1|  date2|dateDiffernece| 
+-------------+-------------+--------------+ 
|1480401142453|1480399932853|    2| 
+-------------+-------------+--------------+ 

,因此你可以得到的区别!

1

因为你有UNIXTIME日期格式,我们可以做这个表达式。

((date1-date2)/(60*60*24*7)).toInt 

编辑: 更新这个答案与例如

spark.udf.register("weekdiff", (from: Long, to: Long) => ((from - to)/(604800)).toInt) 
    // 60*60*24*7 => 604800 
    df.withColumn("weekdiff", weekdiff(df("date1_col_name"), df("date2_col_name")))