Cumulatif

Créez un correctif cumulatif multidimensionnel pour le DataFrame actuel à l’aide des colonnes spécifiées, ce qui permet l’agrégation sur celles-ci.

Syntaxe

rollup(*cols: "ColumnOrNameOrOrdinal")

Paramètres

Paramètre Type Description
cols list, str, int ou Column Colonnes à cumuler. Chaque élément doit être un nom de colonne (chaîne) ou une expression (Column) ou un ordinal de colonne (int, 1) ou une liste d’entre eux.

Retours

GroupedData: données agrégées basées sur les colonnes spécifiées.

Remarques

Un ordinal de colonne commence à partir de 1, qui est différent de celui basé sur __getitem__0.

Exemples

df = spark.createDataFrame([("Alice", 2), ("Bob", 5)], schema=["name", "age"])

df.rollup("name").count().orderBy("name").show()
# +-----+-----+
# | name|count|
# +-----+-----+
# | NULL|    2|
# |Alice|    1|
# |  Bob|    1|
# +-----+-----+

df.rollup("name", df.age).count().orderBy("name", "age").show()
# +-----+----+-----+
# | name| age|count|
# +-----+----+-----+
# | NULL|NULL|    2|
# |Alice|NULL|    1|
# |Alice|   2|    1|
# |  Bob|NULL|    1|
# |  Bob|   5|    1|
# +-----+----+-----+