partitionBy (Fenêtre)

Crée un WindowSpec avec le partitionnement défini.

Syntaxe

Window.partitionBy(*cols)

parameter

parameter	Type	Description
cols	chaîne, Colonne ou liste	Noms de colonnes ou expressions.

parameter	Type	Description
cols	chaîne, Colonne ou liste	Noms de colonnes ou expressions.

Renvoie

WindowSpec

Exemples

Python

from pyspark.sql import Window, functions as sf

df = spark.createDataFrame(
    [(1, "a"), (1, "a"), (2, "a"), (1, "b"), (2, "b"), (3, "b")], ["id", "category"])

# Show row number ordered by id in each category partition.
window = Window.partitionBy("category").orderBy("id")
df.withColumn("row_number", sf.row_number().over(window)).show()
# +---+--------+----------+
# | id|category|row_number|
# +---+--------+----------+
# |  1|       a|         1|
# |  1|       a|         2|
# |  2|       a|         3|
# |  1|       b|         1|
# |  2|       b|         2|
# |  3|       b|         3|
# +---+--------+----------+