Les entiers dans les UDF PySpark
Cet exercice porte sur les UDF, afin de vous aider à comprendre la création de fonctions dans PySpark! Pendant que vous le réalisez, pensez à ce que cela remplacerait dans un flux de nettoyage des données.
N'oubliez pas qu'une SparkSession nommée spark est déjà disponible dans votre espace de travail!
Cette activité fait partie du cours
Introduction à PySpark
Instructions de l’exercice
- Enregistrez la fonction
age_categorycomme UDF sous le nomage_category_udf. - Ajoutez une nouvelle colonne au DataFrame
dfappelée"category"qui applique l'UDF pour catégoriser les personnes selon leur âge. L'argument pourage_category_udf()vous est fourni. - Affichez le DataFrame obtenu.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Register the function age_category as a UDF
age_category_udf = ____(age_category, StringType())
# Apply your udf to the DataFrame
age_category_df_2 = age_category_df.withColumn("category", ____(age_category_df["age"]))
# Show df
age_category_df_2.____