0
我有一个数据帧在unixtime中有两个列日期,我想查找这两列之间的星期差异。 SparkSQL中有一个weekOfYear
UDF,但只有在两个日期都在同一年的时候才有用。那我怎样才能找到周差呢?如何找到两个日期之间的星期差异
p.s.我正在使用Scala Spark。
我有一个数据帧在unixtime中有两个列日期,我想查找这两列之间的星期差异。 SparkSQL中有一个weekOfYear
UDF,但只有在两个日期都在同一年的时候才有用。那我怎样才能找到周差呢?如何找到两个日期之间的星期差异
p.s.我正在使用Scala Spark。
您可以采取的办法为这个创建一个自定义的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|
+-------------+-------------+--------------+
,因此你可以得到的区别!
因为你有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")))
Spark的版本是什么? –
@HZ:考虑接受最好的答案。 – mrsrinivas