Pyspark Converter os resultados da flutuação para substituir inteiro

data_df = data_df.withColumn("Plays", data_df["Plays"].cast(IntegerType()))
Poor Puffin