Let\'s imagine I have the following pyspark dataframe:
data = [("USA",20,40,60), ("India",50,40,30), ("Nepal",20,50,30),