Consider a pyspark dataframe with the following rows with a single column:

1,Australia,5
2,United States,7
3,France,1
5,United Kingdom,10

Write a pyspark function to split each row it so each row is 3 columns with the names 'id', 'country', 'total' and the data in each row appropriately
spam lang_mismatch pii not_appropriate hate_speech sexual_content quality toxicity humor creativity violence