sampleDf = spark.createDataFrame([(1, \'A\', 2021, 1, 5),(1, \'B\', 2021, 1, 6),(1, \'C\', 2021, 1, 7),],[\'msg_id\', \'msg\', \'year\', \'month\', \'day\']) sampleD