I have a PySpark dataframe-
df = spark.createDataFrame([ ("u1", "india", 10), ("u1", "australia", 12), (&q