scala - 如何在 Spark (Scala) 中获取带前导零的整数值

Question

我有 spark 数据框，并尝试向其中添加年、月和日列。但问题是在添加 YTD 列后，它不会在日期和月份列中保留前导零。

val cityDF= Seq(("Delhi","India"),("Kolkata","India"),("Mumbai","India"),("Nairobi","Kenya"),("Colombo","Srilanka"),("Tibet","China")).toDF("City","Country")
val dateString = "2020-01-01"
val dateCol = org.apache.spark.sql.functions.to_date(lit(dateString))
val finaldf = cityDF.select($"*", year(dateCol).alias("Year"), month(dateCol).alias("Month"), dayofmonth(dateCol).alias("Day"))

我想保留月份和日期列中的前导零，但它给我的结果是 1 而不是 01。
因为我使用年月日期列来创建 spark 分区。所以我想保持前导零不变。所以我的问题是:如何在我的数据框列中保留前导零。

Answer 1

整数类型可以转换为字符串类型，前导零是可能的，使用“format_string”函数:

val finaldf =
  cityDF
    .select($"*",
      year(dateCol).alias("Year"),
      format_string("%02d", month(dateCol)).alias("Month"),
      format_string("%02d", dayofmonth(dateCol)).alias("Day")
    )