CommencezCommencez gratuitement

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

Voir le cours

Instructions de l’exercice

  • Enregistrez la fonction age_category comme UDF sous le nom age_category_udf.
  • Ajoutez une nouvelle colonne au DataFrame df appelée "category" qui applique l'UDF pour catégoriser les personnes selon leur âge. L'argument pour age_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.____
Modifier et exécuter le code