Convert date to end of month in Spark

Question

I have a Spark DataFrame as shown below:

#Create DataFrame    
df <- data.frame(name = c("Thomas", "William", "Bill", "John"),
      dates = c('2017-01-05', '2017-02-23', '2017-03-16', '2017-04-08'))
df <- createDataFrame(df)

#Make sure df$dates column is in 'date' format    
df <- withColumn(df, 'dates', cast(df$dates, 'date'))

name    | dates
--------------------
Thomas  |2017-01-05
William |2017-02-23
Bill    |2017-03-16
John    |2017-04-08

I want to change dates to the end of month date, so they would look like shown below. How do I do this? Either SparkR or PySpark code is fine.

name    | dates
--------------------
Thomas  |2017-01-31
William |2017-02-28
Bill    |2017-03-31
John    |2017-04-30

Answer 1

You may use the following (PySpark):

from pyspark.sql.functions import last_day

df.select('name', last_day(df.dates).alias('dates')).show()

To clarify, last_day(date) returns the last day of the month of which date belongs to.

I'm pretty sure there is a similar function in sparkR https://spark.apache.org/docs/1.6.2/api/R/last_day.html

Answer 2

For completeness, here is the SparkR code:

df <- withColumn(df, 'dates', last_day(df$dates))