将天数列添加到 Spark Scala 应用同一数据帧中的日期列

Question

我有一个columns("id", "current_date", "days")dataframedf，我正在尝试将"days"添加到"current_date"，并使用Spark Scala函数创建一个具有新column的新dataframe，称为"new_date"date_add()

val newDF = df.withColumn("new_Date", date_add(df("current_date"), df("days").cast("Int")))

但看起来函数date_add只接受Int值而不是columns.在这种情况下，如何获得所需的输出？我可以使用任何替代函数来获得所需的输出吗？

火花版本：1.6.0 斯卡拉版本：2.10.6

Answer 1

无需使用 UDF，您可以使用 SQL 表达式：

val newDF = df.withColumn("new_date", expr("date_add(current_date,days)"))

Answer 2

可以使用一个小的自定义 udf 来实现此日期算术。

import org.apache.spark.sql.functions.udf
import java.util.concurrent.TimeUnit
import java.util.Date
import java.text.SimpleDateFormat    
val date_add = udf((x: String, y: Int) => {
val sdf = new SimpleDateFormat("yyyy-MM-dd")
val result = new Date(sdf.parse(x).getTime() + TimeUnit.DAYS.toMillis(y))
sdf.format(result)
} )

用法：

scala> val df = Seq((1, "2017-01-01", 10), (2, "2017-01-01", 20)).toDF("id", "current_date", "days")
df: org.apache.spark.sql.DataFrame = [id: int, current_date: string, days: int]
scala> df.withColumn("new_Date", date_add($"current_date", $"days")).show()
+---+------------+----+----------+
| id|current_date|days|  new_Date|
+---+------------+----+----------+
|  1|  2017-01-01|  10|2017-01-11|
|  2|  2017-01-01|  20|2017-01-21|
+---+------------+----+----------+