How to write partitioned json with filename in Spark

Question

I want to write a spark Dataframe into aws s3 as json format.

The dataframe has this columns: id| year| month| day| location_id| data|

I want each